From d391d28191728a83aed80cc3637a2c15d4237a9e Mon Sep 17 00:00:00 2001 From: Your Name Date: Wed, 9 Oct 2024 07:55:18 -0700 Subject: [PATCH] - --- benchmarks/lightning/main.py | 3 ++- docker/Makefile | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/benchmarks/lightning/main.py b/benchmarks/lightning/main.py index 009b28e8..165a0ad2 100644 --- a/benchmarks/lightning/main.py +++ b/benchmarks/lightning/main.py @@ -89,7 +89,8 @@ def main(): # alueError: The `HPUAccelerator` can only be used with a `SingleHPUStrategy` or `HPUParallelStrategy`, found DDPStrategy. - from pytorch_lightning.strategies import HPUParallelStrategy + # from pytorch_lightning.strategies import HPUParallelStrategy + from lightning_habana import HPUParallelStrategy # train model trainer = L.Trainer( diff --git a/docker/Makefile b/docker/Makefile index 92fea4a1..63cbdfd6 100644 --- a/docker/Makefile +++ b/docker/Makefile @@ -10,5 +10,5 @@ hpu: # docker system prune -a -f # docker image prune -a -f docker build --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 + 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 prepare --use-current-env --select lightning && milabench run --use-current-env --select lightning"