Skip to content

Commit 902f942

Browse files
authored
ipex 2.8.10+xpu docker version update (#5733)
1 parent 49804aa commit 902f942

File tree

2 files changed

+3
-7
lines changed

2 files changed

+3
-7
lines changed

docker/Dockerfile.prebuilt

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -95,10 +95,7 @@ ARG IPEX_WHL_URL
9595
RUN python -m pip install numpy \
9696
jinja2==3.1.6
9797

98-
RUN python -m pip install torch==${TORCH_VERSION} --index-url $INDEX_WHL_URL
98+
RUN python -m pip install torch==${TORCH_VERSION} torchvision torchaudio --index-url $INDEX_WHL_URL
9999

100-
RUN python -m pip install torchvision==${TORCHVISION_VERSION} --extra-index-url $NIGHTLY_WHL_URL \
101-
torchaudio==${TORCHAUDIO_VERSION} --extra-index-url $NIGHTLY_WHL_URL
102-
103100
RUN python -m pip install intel_extension_for_pytorch==${IPEX_VERSION} --extra-index-url ${IPEX_WHL_URL} && \
104101
python -m pip install oneccl_bind_pt==${ONECCL_BIND_PT_VERSION} --extra-index-url ${IPEX_WHL_URL}

docker/build.sh

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,11 +15,10 @@ if [[ ${IMAGE_TYPE} = "xpu" ]]; then
1515
--build-arg XPU_SMI_VER=1.2.39-69~22.04 \
1616
--build-arg TORCH_VERSION=2.8.0 \
1717
--build-arg IPEX_VERSION=2.8.10+xpu \
18-
--build-arg TORCHVISION_VERSION=0.23.0.dev20250618+xpu \
19-
--build-arg TORCHAUDIO_VERSION=2.8.0.dev20250618+xpu \
18+
--build-arg TORCHVISION_VERSION=0.23.0+xpu \
19+
--build-arg TORCHAUDIO_VERSION=2.8.0+xpu \
2020
--build-arg ONECCL_BIND_PT_VERSION=2.8.0+xpu \
2121
--build-arg INDEX_WHL_URL=https://download.pytorch.org/whl/test/xpu \
22-
--build-arg NIGHTLY_WHL_URL=https://download.pytorch.org/whl/nightly/xpu \
2322
--build-arg IPEX_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ \
2423
-t ${IMAGE_NAME} \
2524
-f Dockerfile.prebuilt .

0 commit comments

Comments
 (0)