From 943509dae4e7292a5bd176c840ce23e19382331c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9D=92=E9=BE=8D=E8=81=96=E8=80=85=40bdsqlsz?= Date: Tue, 25 Apr 2023 11:38:03 +0800 Subject: [PATCH] Update main.py for return cuda118 --- bitsandbytes_windows/main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/bitsandbytes_windows/main.py b/bitsandbytes_windows/main.py index ae2ad6d8..54ed1eb3 100644 --- a/bitsandbytes_windows/main.py +++ b/bitsandbytes_windows/main.py @@ -410,6 +410,7 @@ def evaluate_cuda_setup(): print(('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')) print('='*80) + return 'libbitsandbytes_cuda118.dll', None, None, None, None if not torch.cuda.is_available(): return 'libbitsandbytes_cpu'+SHARED_LIB_EXTENSION, None, None, None, None cuda_setup = CUDASetup.get_instance()