From 8d87c0b85214c07756b5dcdb09ceb26b0bb1cb7a Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Mon, 31 Oct 2022 18:04:49 -0700 Subject: Fixed CUDA setup bugs, including #81. --- bitsandbytes/cextension.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'bitsandbytes/cextension.py') 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 -- cgit v1.2.3