diff --git a/docker/rocm/Dockerfile b/docker/rocm/Dockerfile index 4aea1276..9b0d19fa 100644 --- a/docker/rocm/Dockerfile +++ b/docker/rocm/Dockerfile @@ -55,7 +55,8 @@ COPY --chown=appuser:appuser docker/rocm/kdb_install.sh /tmp/ RUN /tmp/kdb_install.sh # Support older GFX Arch -RUN cd /tmp && wget https://archlinux.org/packages/extra/x86_64/rocblas/download -O rocblas.tar.zst \ +ENV ROCBLAS_VERSION=6.4.4-1 +RUN cd /tmp && wget https://archive.archlinux.org/packages/r/rocblas/rocblas-${ROCBLAS_VERSION}-x86_64.pkg.tar.zst -O rocblas.tar.zst \ && pwd && ls -lah ./ \ && tar --zstd -xvf rocblas.tar.zst && rm rocblas.tar.zst \ && rm -rf /app/.venv/lib/python3.12/site-packages/torch/lib/rocblas/library/ \ diff --git a/docker/rocm/docker-compose.yml b/docker/rocm/docker-compose.yml index 5a7650b7..8a9fc731 100644 --- a/docker/rocm/docker-compose.yml +++ b/docker/rocm/docker-compose.yml @@ -1,6 +1,8 @@ services: kokoro-tts: - image: kprinssu/kokoro-fastapi:rocm + build: + context: ../.. + dockerfile: docker/rocm/Dockerfile devices: - /dev/dri - /dev/kfd