From bd8b473fff491bcbdb80e83b408805b7db7583a0 Mon Sep 17 00:00:00 2001 From: Your Name Date: Wed, 9 Oct 2024 10:29:20 -0700 Subject: [PATCH] - --- docker/Makefile | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/docker/Makefile b/docker/Makefile index bea14f9e..0f3f24c9 100644 --- a/docker/Makefile +++ b/docker/Makefile @@ -2,7 +2,8 @@ # bench = rlhf-single -bench = lightning -- exclude lightning-gpus +bench = "lightning" +args = "--exclude lightning-gpus" hpu: git add --all @@ -12,5 +13,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 PT_HPU_LAZY_MODE=0 -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)' + docker run -it --runtime=habana -e PT_HPU_LAZY_MODE=0 -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 $(args) --select $(bench)'