Skip to content

Commit 354f8b8

Browse files
authored
fix cuda version
1 parent 645e49f commit 354f8b8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/gui.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def set_cuda_paths():
1919
try:
2020
venv_base = Path(sys.executable).parent
2121
nvidia_base_path = venv_base / 'Lib' / 'site-packages' / 'nvidia'
22-
for env_var in ['CUDA_PATH', 'CUDA_PATH_V12_2']:
22+
for env_var in ['CUDA_PATH', 'CUDA_PATH_V12_1']:
2323
current_path = os.environ.get(env_var, '')
2424
os.environ[env_var] = os.pathsep.join(filter(None, [str(nvidia_base_path), current_path]))
2525
logging.info("CUDA paths set successfully")

0 commit comments

Comments
 (0)