File tree Expand file tree Collapse file tree 1 file changed +6
-9
lines changed
Expand file tree Collapse file tree 1 file changed +6
-9
lines changed Original file line number Diff line number Diff line change @@ -49,18 +49,15 @@ RUN --mount=type=cache,target=/root/.cache/uv \
4949 uv venv --python 3.12 && \
5050 uv sync --extra rocm
5151
52- # Run kbd files
53- ENV ROCM_VERSION=6.4.3
52+ # Support older AMD GPUs
53+ ENV ROCM_VERSION=6.4.4
5454COPY --chown=appuser:appuser docker/rocm/kbd_install.sh /tmp/
55- RUN chmod +x /tmp/kbd_install.sh && /tmp/kbd_install.sh
56-
57- # Support older GFX Arch
58- RUN cd /tmp && wget https://archlinux.org/packages/extra/x86_64/rocblas/download -O rocblas.tar.zst \
59- && pwd && ls -lah ./ \
60- && tar --zstd -xvf rocblas.tar.zst && rm rocblas.tar.zst \
55+ RUN chmod +x /tmp/kbd_install.sh && /tmp/kbd_install.sh \
56+ && cd /tmp && wget https://archlinux.org/packages/extra/x86_64/rocblas/download -O rocblas.tar.zst \
57+ && tar --zstd --keep-old-files --no-overwrite-dir -xvf rocblas.tar.zst && rm rocblas.tar.zst \
6158 && rm -rf /app/.venv/lib/python3.12/site-packages/torch/lib/rocblas/library/ \
6259 && mv ./opt/rocm/lib/rocblas/library/ /app/.venv/lib/python3.12/site-packages/torch/lib/rocblas/ \
63- && rm -rf ./opt && ./usr
60+ && rm -rf /tmp/*
6461
6562# Copy project files including models
6663COPY --chown=appuser:appuser api ./api
You can’t perform that action at this time.
0 commit comments