From 9180b4cc11e6da2936abdcfb3187f35f9e7342fe Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Tue, 3 Jan 2023 06:44:11 -0800 Subject: [PATCH] Added additional error message for cudart error #85 --- bitsandbytes/cuda_setup/main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py index e05e394f9..f99cff6dd 100644 --- a/bitsandbytes/cuda_setup/main.py +++ b/bitsandbytes/cuda_setup/main.py @@ -281,7 +281,12 @@ def get_cuda_version(cuda, cudart_path): return None version = ct.c_int() - check_cuda_result(cuda, cudart.cudaRuntimeGetVersion(ct.byref(version))) + try: + check_cuda_result(cuda, cudart.cudaRuntimeGetVersion(ct.byref(version))) + except AttributeError as e: + CUDASetup.get_instance().add_log_entry(f'ERROR: {str(e)}') + CUDASetup.get_instance().add_log_entry(f'CUDA SETUP: libcudart.so path is {cudart_path}') + CUDASetup.get_instance().add_log_entry(f'CUDA SETUP: Is seems that your cuda installation is not in your path. See https://github.com/TimDettmers/bitsandbytes/issues/85 for more information.') version = int(version.value) major = version//1000 minor = (version-(major*1000))//10