From 7b39a5511d2397914066a2121d757fd4c6530b60 Mon Sep 17 00:00:00 2001 From: blackhc Date: Thu, 29 Dec 2022 23:47:21 +0000 Subject: [PATCH 1/2] Fix issue #97 --- bitsandbytes/cuda_setup/main.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py index 6a6bc74..2b0542d 100644 --- a/bitsandbytes/cuda_setup/main.py +++ b/bitsandbytes/cuda_setup/main.py @@ -143,7 +143,14 @@ def evaluate_cuda_setup(): cuda_setup.add_log_entry(f'CUDA SETUP: Detected CUDA version {cuda_version_string}') # 7.5 is the minimum CC vor cublaslt - has_cublaslt = cc in ["7.5", "8.0", "8.6"] + has_cublaslt = False + if cc is not None: + cc_major, cc_minor = cc.split('.') + if int(cc_major) < 7 or (int(cc_major) == 7 and int(cc_minor) < 5): + failure = True + cuda_setup.add_log_entry("WARNING: Compute capability < 7.5 detected! Proceeding to load CPU-only library...", is_warning=True) + else: + has_cublaslt = True # TODO: # (1) CUDA missing cases (no CUDA installed by CUDA driver (nvidia-smi accessible) From ed17aa9a31a5e38986d596e5fa845a75c3e00264 Mon Sep 17 00:00:00 2001 From: blackhc Date: Thu, 29 Dec 2022 23:50:48 +0000 Subject: [PATCH 2/2] Don't mark it as failure though. --- bitsandbytes/cuda_setup/main.py | 1 - 1 file changed, 1 deletion(-) diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py index 2b0542d..39fee52 100644 --- a/bitsandbytes/cuda_setup/main.py +++ b/bitsandbytes/cuda_setup/main.py @@ -147,7 +147,6 @@ def evaluate_cuda_setup(): if cc is not None: cc_major, cc_minor = cc.split('.') if int(cc_major) < 7 or (int(cc_major) == 7 and int(cc_minor) < 5): - failure = True cuda_setup.add_log_entry("WARNING: Compute capability < 7.5 detected! Proceeding to load CPU-only library...", is_warning=True) else: has_cublaslt = True