Upgrade habana driver to 1.17.1 (#31)

Signed-off-by: kaixuanliu <[email protected]>
huggingface · Aug 26, 2024 · 4334116 · 4334116
1 parent cd26d3d
commit 4334116
Show file tree

Hide file tree

Showing 4 changed files with 289 additions and 264 deletions.
diff --git a/Dockerfile-hpu b/Dockerfile-hpu
@@ -54,7 +54,7 @@ COPY proto proto
 
 RUN cargo build --release --bin text-embeddings-router -F grpc -F python --no-default-features && sccache -s
 
-FROM vault.habana.ai/gaudi-docker/1.17.0/ubuntu22.04/habanalabs/pytorch-installer-2.3.1:latest as base
+FROM vault.habana.ai/gaudi-docker/1.17.1/ubuntu22.04/habanalabs/pytorch-installer-2.3.1:latest as base
 
 ENV HUGGINGFACE_HUB_CACHE=/data \
     PORT=80