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

fix zluda torch patcher

Browse Source
This commit is contained in:
Seunghoon Lee
2024-03-21 11:23:50 +09:00
parent e2cacf8732
commit a0fd8210aa
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+4 -1
View File
@@ -954,7 +954,10 @@ 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_dir = os.environ.get('VENV_DIR', os.path.dirname(shutil.which('python')))
python_dir = os.path.dirname(shutil.which('python'))
if shutil.which('conda') is None:
python_dir = os.path.dirname(python_dir)
venv_dir = os.environ.get('VENV_DIR', python_dir)
dlls_to_patch = {
'cublas.dll': 'cublas64_11.dll',
#'cudnn.dll': 'cudnn64_8.dll',