Added better error message for debugging on CUDA not detected failures.
This commit is contained in:
parent
659a7dfc71
commit
7c651012fc
|
@ -18,16 +18,20 @@ try:
|
|||
CUDASetup.get_instance().generate_instructions()
|
||||
CUDASetup.get_instance().print_log_stack()
|
||||
raise RuntimeError('''
|
||||
CUDA Setup failed despite GPU being available. Inspect the CUDA SETUP outputs above to fix your environment!
|
||||
If you cannot find any issues and suspect a bug, please open an issue with detals about your environment:
|
||||
https://github.com/TimDettmers/bitsandbytes/issues''')
|
||||
CUDA Setup failed despite GPU being available. Please run the following command to get more information:
|
||||
|
||||
python -m bitsandbytes
|
||||
|
||||
Inspect the output of the command and see if you can locate CUDA libraries. You might need to add them
|
||||
to your LD_LIBRARY_PATH. If you suspect a bug, please take the information from python -m bitsandbytes
|
||||
and open an issue at: https://github.com/TimDettmers/bitsandbytes/issues''')
|
||||
lib.cadam32bit_g32
|
||||
lib.get_context.restype = ct.c_void_p
|
||||
lib.get_cusparse.restype = ct.c_void_p
|
||||
COMPILED_WITH_CUDA = True
|
||||
except AttributeError:
|
||||
warn("The installed version of bitsandbytes was compiled without GPU support. "
|
||||
"8-bit optimizers and GPU quantization are unavailable.")
|
||||
"8-bit optimizers, 8-bit multiplication, and GPU quantization are unavailable.")
|
||||
COMPILED_WITH_CUDA = False
|
||||
|
||||
# print the setup details after checking for errors so we do not print twice
|
||||
|
|
Loading…
Reference in New Issue
Block a user