-

mila-iqia · Oct 9, 2024 · e1cac26 · e1cac26
1 parent c7327a3
commit e1cac26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Makefile b/docker/Makefile
@@ -11,5 +11,5 @@ hpu:
 	# docker system prune -a -f
 	# docker image prune -a -f
 	docker build --build-arg BENCH=$(bench) --build-arg CACHEBUST=`git rev-parse hpu` -f Dockerfile-hpu -t dockerfile-hpu . 
-	docker run   -it   --runtime=habana   -e HABANA_VISIBLE_DEVICES=all   -e OMPI_MCA_btl_vader_single_copy_mechanism=none --shm-size 50G   --cap-add=sys_nice   --net=host   dockerfile-hpu:latest bash -c '. $$MILABENCH_VENV/bin/activate && milabench install --use-current-env --select $(bench) &&  milabench prepare --use-current-env --select $(bench) && milabench run --use-current-env --select $(bench)'
+	docker run   -it   --runtime=habana   -e HABANA_VISIBLE_DEVICES=all   -e OMPI_MCA_btl_vader_single_copy_mechanism=none --shm-size 50G   --cap-add=sys_nice   --net=host   dockerfile-hpu:latest bash -c '. $$MILABENCH_VENV/bin/activate && milabench install --use-current-env --select $(bench) && pip uninstall torch torchvision torchaudio -y  &&  milabench prepare --use-current-env --select $(bench) && milabench run --use-current-env --select $(bench)'