Skip to content

Commit 7c65101

Browse files
committed
Added better error message for debugging on CUDA not detected failures.
1 parent 659a7df commit 7c65101

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

bitsandbytes/cextension.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,16 +18,20 @@
1818
CUDASetup.get_instance().generate_instructions()
1919
CUDASetup.get_instance().print_log_stack()
2020
raise RuntimeError('''
21-
CUDA Setup failed despite GPU being available. Inspect the CUDA SETUP outputs above to fix your environment!
22-
If you cannot find any issues and suspect a bug, please open an issue with detals about your environment:
23-
https://github.com/TimDettmers/bitsandbytes/issues''')
21+
CUDA Setup failed despite GPU being available. Please run the following command to get more information:
22+
23+
python -m bitsandbytes
24+
25+
Inspect the output of the command and see if you can locate CUDA libraries. You might need to add them
26+
to your LD_LIBRARY_PATH. If you suspect a bug, please take the information from python -m bitsandbytes
27+
and open an issue at: https://github.com/TimDettmers/bitsandbytes/issues''')
2428
lib.cadam32bit_g32
2529
lib.get_context.restype = ct.c_void_p
2630
lib.get_cusparse.restype = ct.c_void_p
2731
COMPILED_WITH_CUDA = True
2832
except AttributeError:
2933
warn("The installed version of bitsandbytes was compiled without GPU support. "
30-
"8-bit optimizers and GPU quantization are unavailable.")
34+
"8-bit optimizers, 8-bit multiplication, and GPU quantization are unavailable.")
3135
COMPILED_WITH_CUDA = False
3236

3337
# print the setup details after checking for errors so we do not print twice

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def read(fname):
1818

1919
setup(
2020
name=f"bitsandbytes",
21-
version=f"0.38.0",
21+
version=f"0.38.0.post2",
2222
author="Tim Dettmers",
2323
author_email="dettmers@cs.washington.edu",
2424
description="8-bit optimizers and matrix multiplication routines.",

0 commit comments

Comments
 (0)