From 26efb154c8d77b4ede2cfc0dbd2381dd385f33e7 Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Sun, 7 Aug 2022 09:49:36 -0700 Subject: Fixed bug where python -m bitsandbytes was failing. --- bitsandbytes/__main__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'bitsandbytes') diff --git a/bitsandbytes/__main__.py b/bitsandbytes/__main__.py index 7f3d24c..5f11875 100644 --- a/bitsandbytes/__main__.py +++ b/bitsandbytes/__main__.py @@ -30,7 +30,7 @@ print() from . import COMPILED_WITH_CUDA, PACKAGE_GITHUB_URL -from .cuda_setup.main import get_compute_capabilities +from .cuda_setup.main import get_compute_capabilities, get_cuda_lib_handle from .cuda_setup.env_vars import to_be_ignored from .utils import print_stderr @@ -47,7 +47,8 @@ print( print_header("OTHER") print(f"{COMPILED_WITH_CUDA = }") -print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities()}") +cuda = get_cuda_lib_handle() +print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities(cuda)}") print_header("") print_header("DEBUG INFO END") print_header("") -- cgit v1.2.3