Merge pull request #595 from ihsanturk/FIX-__main__.py-REFERENCE-TO-NONEXISTENT-get_cuda_lib_handle

Fix  import crash caused by __main__.py reference to nonexistent cuda_setup.main.get_cuda_lib_handle

Author: TimDettmers

bitsandbytes/__main__.py

@@ -97,13 +97,12 @@ def print_debug_info() -> None:
 
 from . import COMPILED_WITH_CUDA, PACKAGE_GITHUB_URL
 from .cuda_setup.env_vars import to_be_ignored
-from .cuda_setup.main import get_compute_capabilities, get_cuda_lib_handle
+from .cuda_setup.main import get_compute_capabilities
 
 
 print_header("OTHER")
 print(f"COMPILED_WITH_CUDA = {COMPILED_WITH_CUDA}")
-cuda = get_cuda_lib_handle()
-print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities(cuda)}")
+print(f"COMPUTE_CAPABILITIES_PER_GPU = {get_compute_capabilities()}")
 print_header("")
 print_header("DEBUG INFO END")
 print_header("")

bitsandbytes/cuda_setup/main.py

@@ -359,4 +359,4 @@ def evaluate_cuda_setup():
         "if not has_cublaslt (CC < 7.5), then we have to choose  _nocublaslt.so"
         binary_name = f"libbitsandbytes_cuda{cuda_version_string}_nocublaslt.so"
 
-    return binary_name, cudart_path, cc, cuda_version_string
+    return binary_name, cudart_path, cc, cuda_version_string