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

fix: fix tensorflow installer for navi 3x

Browse Source
This commit is contained in:
evshiron
2023-08-14 00:57:40 +08:00
parent b3029200f5
commit bb6b3e2e3f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+2 -1
View File
@@ -344,7 +344,8 @@ def check_torch():
if arch == 'navi3x':
os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', '11.0.0')
# do not use tensorflow-rocm for navi 3x
os.environ.setdefault('TENSORFLOW_PACKAGE', 'tensorflow==2.13.0')
if os.environ.get('TENSORFLOW_PACKAGE') == 'tensorflow-rocm':
os.environ['TENSORFLOW_PACKAGE'] = 'tensorflow==2.13.0'
elif arch == 'navi2x':
os.environ.setdefault('HSA_OVERRIDE_GFX_VERSION', '10.3.0')
else: