diff options
| -rw-r--r-- | bitsandbytes/cuda_setup.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/bitsandbytes/cuda_setup.py b/bitsandbytes/cuda_setup.py index 08f3ed2..2fb5049 100644 --- a/bitsandbytes/cuda_setup.py +++ b/bitsandbytes/cuda_setup.py @@ -131,8 +131,7 @@ def evaluate_cuda_setup(): cc = get_compute_capability() binary_name = 'libbitsandbytes_cpu.so' - has_gpu = cc != '' - if not has_gpu: + if not (has_gpu := bool(cc)): print('WARNING: No GPU detected! Check our CUDA paths. Processing to load CPU-only library...') return binary_name |