Make light ROCm containers

mila-iqia · Aug 5, 2024 · 1b80d4f · 1b80d4f
1 parent 886d2a3
commit 1b80d4f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/docker/Dockerfile-rocm b/docker/Dockerfile-rocm
@@ -70,7 +70,6 @@ RUN python -m pip install -U pip            &&\
 ENV PIP_DEFAULT_TIMEOUT=800
 
 RUN milabench install --config $MILABENCH_CONFIG --base $MILABENCH_BASE $MILABENCH_ARGS &&\
-    milabench prepare --config $MILABENCH_CONFIG --base $MILABENCH_BASE $MILABENCH_ARGS &&\
     python -m pip cache purge
 
 CMD ["milabench", "run"]