Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use TensorFlow ITEX for Intel GPUs

Browse Source
This commit is contained in:
Disty0
2023-07-23 15:47:15 +03:00
parent fae6033abc
commit 86099dfb15
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+1
View File
@@ -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()