update dockerfile

Signed-off-by: Sun, Xuehao <[email protected]>
opea-project · Nov 6, 2024 · 1232d68 · 1232d68
1 parent 47bb4ac
commit 1232d68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/hpu.dockerfile b/docker/hpu.dockerfile
@@ -1,4 +1,4 @@
-FROM vault.habana.ai/gaudi-docker/1.16.1/ubuntu22.04/habanalabs/pytorch-installer-2.2.2:latest as hpu
+FROM vault.habana.ai/gaudi-docker/1.18.0/ubuntu22.04/habanalabs/pytorch-installer-2.4.0:latest as hpu
 
 ENV LANG=en_US.UTF-8
 ENV PYTHONPATH=/root:/usr/lib/habanalabs/