We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 32491bc commit 6dae722Copy full SHA for 6dae722
bitsandbytes/cuda_setup/main.py
@@ -33,9 +33,7 @@
33
# we have libcudart.so.11.0 which causes a lot of errors before
34
# not sure if libcudart.so.12.0 exists in pytorch installs, but it does not hurt
35
system = platform.system()
36
-if system == 'Darwin':
37
- CUDA_RUNTIME_LIBS: list = ["libcuda.dylib", '/usr/local/cuda/lib/libcuda.dylib']
38
-elif system == 'Windows':
+if system == 'Windows':
39
CUDA_RUNTIME_LIBS: list = ["nvcuda.dll"]
40
else: # Linux or other
41
CUDA_RUNTIME_LIBS: list = ["libcudart.so", 'libcudart.so.11.0', 'libcudart.so.12.0']
0 commit comments