diff options
| author | Tim Dettmers <tim.dettmers@gmail.com> | 2022-10-31 18:04:49 -0700 |
|---|---|---|
| committer | Tim Dettmers <tim.dettmers@gmail.com> | 2022-10-31 18:04:49 -0700 |
| commit | 8d87c0b85214c07756b5dcdb09ceb26b0bb1cb7a (patch) | |
| tree | 197e95b4fa2bcd59300c60402e6f4568608f8484 /bitsandbytes/cextension.py | |
| parent | 4844aef4ff51b433d549a9acc49b11a925b6f4fe (diff) | |
Fixed CUDA setup bugs, including #81.
Diffstat (limited to 'bitsandbytes/cextension.py')
| -rw-r--r-- | bitsandbytes/cextension.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/bitsandbytes/cextension.py b/bitsandbytes/cextension.py index 4f791e2..8125202 100644 --- a/bitsandbytes/cextension.py +++ b/bitsandbytes/cextension.py @@ -116,7 +116,7 @@ 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 to fix your environment! + CUDA Setup failed despite GPU being available. Inspect the CUDA SETUP outputs aboveto 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''') lib.cadam32bit_g32 @@ -124,8 +124,6 @@ try: 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." - ) + warn("The installed version of bitsandbytes was compiled without GPU support. " + "8-bit optimizers and GPU quantization are unavailable.") COMPILED_WITH_CUDA = False |