Skip to content

Commit 39f9599

Browse files
committed
Clean up the older gfx build step to minimize image size
1 parent 8f2b543 commit 39f9599

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

docker/rocm/Dockerfile

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff 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
5454
COPY --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
6663
COPY --chown=appuser:appuser api ./api

0 commit comments

Comments
 (0)