Skip to content

Commit 0488566

Browse files
authored
fix cudart*dll for Windows (#1064)
1 parent 1d709aa commit 0488566

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bitsandbytes/cuda_setup/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030

3131
DYNAMIC_LIBRARY_SUFFIX = { "Darwin": ".dylib", "Windows": ".dll", "Linux": ".so"}.get(platform.system(), ".so")
3232
if platform.system() == "Windows": # Windows
33-
CUDA_RUNTIME_LIBS = ["nvcuda.dll"]
33+
CUDA_RUNTIME_LIBS = ["cudart64_110.dll", "cudart64_12.dll"]
3434
else: # Linux or other
3535
# these are the most common libs names
3636
# libcudart.so is missing by default for a conda install with PyTorch 2.0 and instead

0 commit comments

Comments
 (0)