From e1cac263b3fd968c91d9a78833a84e21df888f9f Mon Sep 17 00:00:00 2001 From: Your Name Date: Wed, 9 Oct 2024 09:05:37 -0700 Subject: [PATCH] - --- docker/Makefile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Makefile b/docker/Makefile index e2eeac30..d149f4e4 100644 --- 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)'