File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -96,6 +96,6 @@ RUN python -m pip install numpy \
96
96
jinja2==3.1.6
97
97
98
98
RUN python -m pip install torch==${TORCH_VERSION} torchvision torchaudio --index-url $INDEX_WHL_URL
99
-
100
- RUN python -m pip install intel_extension_for_pytorch==${IPEX_VERSION} --extra-index-url ${IPEX_WHL_URL} && \
101
- python -m pip install oneccl_bind_pt==${ONECCL_BIND_PT_VERSION} --extra-index-url ${IPEX_WHL_URL}
99
+
100
+ RUN python -m pip install intel_extension_for_pytorch==${IPEX_VERSION} \
101
+ oneccl_bind_pt==${ONECCL_BIND_PT_VERSION} --extra-index-url ${IPEX_WHL_URL}
Original file line number Diff line number Diff line change @@ -18,7 +18,7 @@ if [[ ${IMAGE_TYPE} = "xpu" ]]; then
18
18
--build-arg TORCHVISION_VERSION=0.23.0+xpu \
19
19
--build-arg TORCHAUDIO_VERSION=2.8.0+xpu \
20
20
--build-arg ONECCL_BIND_PT_VERSION=2.8.0+xpu \
21
- --build-arg INDEX_WHL_URL=https://download.pytorch.org/whl/test/ xpu \
21
+ --build-arg INDEX_WHL_URL=https://download.pytorch.org/whl/xpu \
22
22
--build-arg IPEX_WHL_URL=https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/ \
23
23
-t ${IMAGE_NAME} \
24
24
-f Dockerfile.prebuilt .
You can’t perform that action at this time.
0 commit comments