From f5cab1185d45e08f9996b279eb0edd2ff06a8619 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Thu, 14 Dec 2023 12:13:28 +0300 Subject: [PATCH] IPEX update Tensorflow --- installer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/installer.py b/installer.py index c6725287b..fb0f462ed 100644 --- a/installer.py +++ b/installer.py @@ -424,7 +424,8 @@ def check_torch(): os.environ.setdefault('NEOReadDebugKeys', '1') os.environ.setdefault('ClDeviceGlobalMemSizeAvailablePercent', '100') torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.1.0a0 torchvision==0.16.0a0 intel-extension-for-pytorch==2.1.10+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/') - os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow==2.13.0 intel-extension-for-tensorflow[gpu]') + if "linux" in sys.platform: + os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow==2.14.0 intel-extension-for-tensorflow[xpu]==2.14.0.1') install('mkl==2024.0.0', 'mkl') install('mkl-dpcpp==2024.0.0', 'mkl-dpcpp') install('openvino', 'openvino', ignore=True)