Skip to content

GitLab

Unverified Commit 04885664 authored by Won-Kyu Park's avatar Won-Kyu Park Committed by GitHub
Browse files

fix cudart*dll for Windows (#1064)

parent 1d709aad
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
bitsandbytes/cuda_setup/main.py
View file @ 04885664
...@@ -30,7 +30,7 @@ from .env_vars import get_potentially_lib_path_containing_env_vars ...@@ -30,7 +30,7 @@ from .env_vars import get_potentially_lib_path_containing_env_vars
DYNAMIC_LIBRARY_SUFFIX = { "Darwin": ".dylib", "Windows": ".dll", "Linux": ".so"}.get(platform.system(), ".so") DYNAMIC_LIBRARY_SUFFIX = { "Darwin": ".dylib", "Windows": ".dll", "Linux": ".so"}.get(platform.system(), ".so")
if platform.system() == "Windows": # Windows if platform.system() == "Windows": # Windows
CUDA_RUNTIME_LIBS = ["nvcuda.dll"] CUDA_RUNTIME_LIBS = ["cudart64_110.dll", "cudart64_12.dll"]
else: # Linux or other else: # Linux or other
# these are the most common libs names # these are the most common libs names
# libcudart.so is missing by default for a conda install with PyTorch 2.0 and instead # libcudart.so is missing by default for a conda install with PyTorch 2.0 and instead
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment