Skip to content

Commit 80d3b02

Browse files
Upgrade Pytorch for XPU (#3366)
* upgraded the versions of some dependencies * added onnxruntime to requirements_linux_ipex.txt * upgraded the versions of torch and torchvision * downgraded the torch version to match the official image
1 parent e794bb8 commit 80d3b02

File tree

1 file changed

+13
-10
lines changed

1 file changed

+13
-10
lines changed

requirements_linux_ipex.txt

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,22 @@
11
# Custom index URL for specific packages
2+
--extra-index-url https://download.pytorch.org/whl/xpu
23
--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
34

4-
torch==2.3.1+cxx11.abi
5-
torchvision==0.18.1+cxx11.abi
6-
intel-extension-for-pytorch==2.3.110+xpu
7-
oneccl_bind_pt==2.3.100+xpu
5+
torch==2.7.0+xpu
6+
torchvision==0.22.0+xpu
7+
intel-extension-for-pytorch==2.7.10+xpu
8+
oneccl_bind_pt==2.7.0+xpu
89

910
tensorboard==2.15.2
10-
tensorflow==2.15.1
11-
intel-extension-for-tensorflow[xpu]==2.15.0.1
11+
tensorflow==2.15.0
12+
intel-extension-for-tensorflow[xpu]==2.15.0.2
13+
onnxruntime==1.22.1
1214
onnxruntime-openvino==1.22.0
1315

14-
mkl==2024.2.1
15-
mkl-dpcpp==2024.2.1
16-
oneccl-devel==2021.13.1
17-
impi-devel==2021.13.1
16+
mkl==2025.0.1
17+
mkl-dpcpp==2025.0.1
18+
oneccl-devel==2021.14.1
19+
impi-devel==2021.14.1
1820

1921
-r requirements.txt
22+

0 commit comments

Comments
 (0)