-

mila-iqia · Oct 9, 2024 · d391d28 · d391d28
1 parent c419535
commit d391d28
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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"