diff --git a/Dockerfile-intel b/Dockerfile-intel index 695ae17e..47c5b165 100644 --- a/Dockerfile-intel +++ b/Dockerfile-intel @@ -82,7 +82,7 @@ RUN python -m pip install torch==2.7.0 torchvision torchaudio==2.7.0 --index-url RUN cd backends/python/server && \ make install -FROM vault.habana.ai/gaudi-docker/1.21.0/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest AS hpu +FROM vault.habana.ai/gaudi-docker/1.21.3/ubuntu22.04/habanalabs/pytorch-installer-2.6.0:latest AS hpu ENV HUGGINGFACE_HUB_CACHE=/data \ PORT=80 ENV PT_HPU_LAZY_MODE=1