Skip to content

Commit 8f2b543

Browse files
committed
Ensure that extra downloaded rocBlas files are cleaned up
1 parent 7b14be6 commit 8f2b543

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

docker/rocm/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,8 @@ RUN cd /tmp && wget https://archlinux.org/packages/extra/x86_64/rocblas/download
5959
&& pwd && ls -lah ./ \
6060
&& tar --zstd -xvf rocblas.tar.zst && rm rocblas.tar.zst \
6161
&& rm -rf /app/.venv/lib/python3.12/site-packages/torch/lib/rocblas/library/ \
62-
&& mv ./opt/rocm/lib/rocblas/library/ /app/.venv/lib/python3.12/site-packages/torch/lib/rocblas/
62+
&& mv ./opt/rocm/lib/rocblas/library/ /app/.venv/lib/python3.12/site-packages/torch/lib/rocblas/ \
63+
&& rm -rf ./opt && ./usr
6364

6465
# Copy project files including models
6566
COPY --chown=appuser:appuser api ./api

0 commit comments

Comments
 (0)