From 74081b96280ea1ac1972e523475166d08e81873d Mon Sep 17 00:00:00 2001 From: ChengyuZhu6 Date: Fri, 16 Aug 2024 09:39:42 +0800 Subject: [PATCH] docker: Update prebuilt Dockerfile 1. Bump pytorch and ipex to 2.4.0 2. update the link of `TORCH_CPU_URL`. `https://download.pytorch.org/whl/cpu/torch_stable.html` is deprecated since torch 2.4. See https://github.com/pytorch/pytorch/issues/133128. Signed-off-by: ChengyuZhu6 --- docker/Dockerfile.prebuilt | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/docker/Dockerfile.prebuilt b/docker/Dockerfile.prebuilt index 5d2275355..501a38afb 100644 --- a/docker/Dockerfile.prebuilt +++ b/docker/Dockerfile.prebuilt @@ -28,15 +28,15 @@ RUN ${PYTHON} -m pip --no-cache-dir install --upgrade \ # Some TF tools expect a "python" binary RUN ln -s $(which ${PYTHON}) /usr/local/bin/python -ARG IPEX_VERSION=2.3.100 -ARG PYTORCH_VERSION=2.3.0 -ARG TORCHAUDIO_VERSION=2.3.0 -ARG TORCHVISION_VERSION=0.18.0 -ARG TORCH_CPU_URL=https://download.pytorch.org/whl/cpu/torch_stable.html +ARG IPEX_VERSION=2.4.0 +ARG PYTORCH_VERSION=2.4.0 +ARG TORCHAUDIO_VERSION=2.4.0 +ARG TORCHVISION_VERSION=0.19.0 +ARG TORCH_CPU_URL=https://download.pytorch.org/whl/cpu/ RUN \ - python -m pip install --no-cache-dir \ - torch==${PYTORCH_VERSION}+cpu torchvision==${TORCHVISION_VERSION}+cpu torchaudio==${TORCHAUDIO_VERSION}+cpu -f ${TORCH_CPU_URL} && \ + python -m pip install --extra-index-url ${TORCH_CPU_URL} --no-cache-dir \ + torch==${PYTORCH_VERSION}+cpu torchvision==${TORCHVISION_VERSION}+cpu torchaudio==${TORCHAUDIO_VERSION}+cpu && \ python -m pip install --no-cache-dir \ intel_extension_for_pytorch==${IPEX_VERSION}