Skip to content

Commit 6e97421

Browse files
committed
IPEX update to 2.1.40+xpu
1 parent dcedcba commit 6e97421

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

installer.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -546,13 +546,13 @@ def install_ipex(torch_command):
546546
if os.environ.get("ClDeviceGlobalMemSizeAvailablePercent", None) is None:
547547
os.environ.setdefault('ClDeviceGlobalMemSizeAvailablePercent', '100')
548548
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/')
550550
# os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow==2.15.0 intel-extension-for-tensorflow[xpu]==2.15.0.0')
551551
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')
556556
else:
557557
if sys.version_info.minor == 11:
558558
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

Comments
 (0)