From 86099dfb15c6f57dec24bcb55fa7a3fe89098221 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Sun, 23 Jul 2023 15:47:15 +0300 Subject: [PATCH] Use TensorFlow ITEX for Intel GPUs --- installer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/installer.py b/installer.py index 1e7bda491..cbb60c7fd 100644 --- a/installer.py +++ b/installer.py @@ -322,6 +322,7 @@ def check_torch(): log.error('Intel OneAPI Toolkit is not activated! Start the WebUI with --use-ipex or activate OneAPI manually') os.environ.setdefault('NEOReadDebugKeys', '1') os.environ.setdefault('ClDeviceGlobalMemSizeAvailablePercent', '100') + os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow==2.12.0 intel-extension-for-tensorflow[gpu]') torch_command = os.environ.get('TORCH_COMMAND', 'torch==1.13.0a0+git6c9b55e torchvision==0.14.1a0 intel_extension_for_pytorch==1.13.120+xpu -f https://developer.intel.com/ipex-whl-stable-xpu') else: machine = platform.machine()