diff --git a/bitsandbytes/__main__.py b/bitsandbytes/__main__.py index a291fab..523d023 100644 --- a/bitsandbytes/__main__.py +++ b/bitsandbytes/__main__.py @@ -98,13 +98,12 @@ generate_bug_report_information() 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("") diff --git a/bitsandbytes/cuda_setup/main.py b/bitsandbytes/cuda_setup/main.py index 1aaf9a0..c1dfce2 100644 --- a/bitsandbytes/cuda_setup/main.py +++ b/bitsandbytes/cuda_setup/main.py @@ -87,7 +87,7 @@ class CUDASetup: make_cmd += '_nomatmul' self.add_log_entry('CUDA SETUP: Something unexpected happened. Please compile from source:') - self.add_log_entry('git clone git@github.com:TimDettmers/bitsandbytes.git') + self.add_log_entry('git clone https://github.com/TimDettmers/bitsandbytes.git') self.add_log_entry('cd bitsandbytes') self.add_log_entry(make_cmd) self.add_log_entry('python setup.py install') @@ -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 \ No newline at end of file