File tree Expand file tree Collapse file tree 2 files changed +3
-7
lines changed
Expand file tree Collapse file tree 2 files changed +3
-7
lines changed Original file line number Diff line number Diff line change @@ -95,10 +95,7 @@ ARG IPEX_WHL_URL
9595RUN 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-
103100RUN 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}
Original file line number Diff line number Diff 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 .
You can’t perform that action at this time.
0 commit comments