Skip to content

Commit 83f643c

Browse files
authored
Include x64 directory for CUDA DLLs on Windows
1 parent c37132b commit 83f643c

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

llama_cpp/_ctypes_extensions.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ def load_shared_library(lib_base_name: str, base_path: pathlib.Path):
5454
os.add_dll_directory(str(base_path))
5555
if "CUDA_PATH" in os.environ:
5656
os.add_dll_directory(os.path.join(os.environ["CUDA_PATH"], "bin"))
57+
os.add_dll_directory(os.path.join(os.environ["CUDA_PATH"], "bin", "x64"))
5758
os.add_dll_directory(os.path.join(os.environ["CUDA_PATH"], "lib"))
5859
if "HIP_PATH" in os.environ:
5960
os.add_dll_directory(os.path.join(os.environ["HIP_PATH"], "bin"))

0 commit comments

Comments
 (0)