@@ -546,13 +546,13 @@ def install_ipex(torch_command):
546
546
if os .environ .get ("ClDeviceGlobalMemSizeAvailablePercent" , None ) is None :
547
547
os .environ .setdefault ('ClDeviceGlobalMemSizeAvailablePercent' , '100' )
548
548
if "linux" in sys .platform :
549
- torch_command = os .environ .get ('TORCH_COMMAND' , 'torch==2.1.0.post0 torchvision==0.16.0.post0 intel-extension-for-pytorch==2.1.20 +xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/' )
549
+ torch_command = os .environ .get ('TORCH_COMMAND' , 'torch==2.1.0.post3 torchvision==0.16.0.post3 intel-extension-for-pytorch==2.1.40+xpu oneccl_bind_pt==2.1.400 +xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/' )
550
550
# os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow==2.15.0 intel-extension-for-tensorflow[xpu]==2.15.0.0')
551
551
if os .environ .get ('DISABLE_VENV_LIBS' , None ) is None :
552
- install (os .environ .get ('MKL_PACKAGE' , 'mkl==2024.1 .0' ), 'mkl' )
553
- install (os .environ .get ('DPCPP_PACKAGE' , 'mkl-dpcpp==2024.1 .0' ), 'mkl-dpcpp' )
554
- install (os .environ .get ('ONECCL_PACKAGE' , 'oneccl-devel==2021.12 .0' ), 'oneccl-devel' )
555
- install (os .environ .get ('MPI_PACKAGE' , 'impi-devel==2021.12 .0' ), 'impi-devel' )
552
+ install (os .environ .get ('MKL_PACKAGE' , 'mkl==2024.2 .0' ), 'mkl' )
553
+ install (os .environ .get ('DPCPP_PACKAGE' , 'mkl-dpcpp==2024.2 .0' ), 'mkl-dpcpp' )
554
+ install (os .environ .get ('ONECCL_PACKAGE' , 'oneccl-devel==2021.13 .0' ), 'oneccl-devel' )
555
+ install (os .environ .get ('MPI_PACKAGE' , 'impi-devel==2021.13 .0' ), 'impi-devel' )
556
556
else :
557
557
if sys .version_info .minor == 11 :
558
558
pytorch_pip = 'https://github.com/Nuullll/intel-extension-for-pytorch/releases/download/v2.1.10%2Bxpu/torch-2.1.0a0+cxx11.abi-cp311-cp311-win_amd64.whl'
0 commit comments