1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
dev
Davi Henrique Linhares 2025-07-26 05:01:06 -03:00 committed by GitHub
parent e794bb8a5d
commit 80d3b021a9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
requirements_linux_ipex.txt
View File

@ -1,19 +1,22 @@
# Custom index URL for specific packages
--extra-index-url https://download.pytorch.org/whl/xpu
--extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
torch==2.3.1+cxx11.abi
torchvision==0.18.1+cxx11.abi
intel-extension-for-pytorch==2.3.110+xpu
oneccl_bind_pt==2.3.100+xpu
torch==2.7.0+xpu
torchvision==0.22.0+xpu
intel-extension-for-pytorch==2.7.10+xpu
oneccl_bind_pt==2.7.0+xpu
tensorboard==2.15.2
tensorflow==2.15.1
intel-extension-for-tensorflow[xpu]==2.15.0.1
tensorflow==2.15.0
intel-extension-for-tensorflow[xpu]==2.15.0.2
onnxruntime==1.22.1
onnxruntime-openvino==1.22.0
mkl==2024.2.1
mkl-dpcpp==2024.2.1
oneccl-devel==2021.13.1
impi-devel==2021.13.1
mkl==2025.0.1
mkl-dpcpp==2025.0.1
oneccl-devel==2021.14.1
impi-devel==2021.14.1
-r requirements.txt