Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-09-05 20:41:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

Why not search for cuda_path as well? (#865)

Browse Source 
Let's add dll directory for cuda on Windows too.
This commit is contained in:
jojorne
2024-05-27 10:38:17 -03:00
committed by GitHub
parent 318d5b87fc
commit dc53e30785
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
koboldcpp.py
+2
View File
@@ -275,6 +275,8 @@ def init_library():
os.add_dll_directory(dir_path)
os.add_dll_directory(abs_path)
os.add_dll_directory(os.getcwd())
if libname == lib_cublas and "CUDA_PATH" in os.environ:
os.add_dll_directory(os.path.join(os.environ["CUDA_PATH"], "bin"))
if libname == lib_hipblas and "HIP_PATH" in os.environ:
os.add_dll_directory(os.path.join(os.environ["HIP_PATH"], "bin"))
if args.debugmode == 1: