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

fix installer for zluda

Browse Source
This commit is contained in:
Seunghoon Lee
2024-02-24 13:35:22 +09:00
committed by Vladimir Mandic
parent b32dbf16a4
commit 72e938621b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+2 -2
View File
@@ -920,7 +920,7 @@ def patch_zluda():
if zluda_path is None:
log.warning('Failed to automatically patch torch with ZLUDA. Could not find ZLUDA from PATH.')
return
venv_path = os.path.dirname(shutil.which('python'))
venv_dir = os.environ.get('VENV_DIR', os.path.dirname(shutil.which('python')))
dlls_to_patch = {
'cublas.dll': 'cublas64_11.dll',
'cudnn.dll': 'cudnn64_8.dll',
@@ -929,7 +929,7 @@ def patch_zluda():
}
try:
for k, v in dlls_to_patch.items():
shutil.copyfile(os.path.join(zluda_path, k), os.path.join(venv_path, 'Lib', 'site-packages', 'torch', 'lib', v))
shutil.copyfile(os.path.join(zluda_path, k), os.path.join(venv_dir, 'Lib', 'site-packages', 'torch', 'lib', v))
except Exception as e:
log.warning(f'ZLUDA: failed to automatically patch torch: {e}')