This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Dockerfile for building CTranslate2 with ROCm support for Python 3.10 | |
# Tested with ROCm 6.2 and AMD RX 7900 XT (gfx1100) | |
# Source: https://github.com/arlo-phoenix/CTranslate2-rocm | |
FROM rocm/pytorch | |
# Set working directory for build | |
WORKDIR /build | |
# Activate conda environment for the rest of the commands |