mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #593 from bilelomrani1/main

Browse Source 
Fix bitsandbytes import error when CUDA is unavailable
This commit is contained in:
Tim Dettmers 2023-07-14 17:47:48 -07:00 committed by GitHub
commit 8cdec888b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bitsandbytes/cuda_setup/main.py
View File

@ -330,7 +330,7 @@ def evaluate_cuda_setup():
cuda_setup.add_log_entry(('Welcome to bitsandbytes. For bug reports, please run\n\npython -m bitsandbytes\n\n'),
('and submit this information together with your error trace to: https://github.com/TimDettmers/bitsandbytes/issues'))
cuda_setup.add_log_entry('='*80)
if not torch.cuda.is_available(): return 'libbitsandbytes_cpu.so', None, None, None, None
if not torch.cuda.is_available(): return 'libbitsandbytes_cpu.so', None, None, None
cudart_path = determine_cuda_runtime_lib_path()
ccs = get_compute_capabilities()