Skip to content

GitLab

Commit 26efb154 authored by Tim Dettmers's avatar Tim Dettmers
Browse files

Fixed bug where python -m bitsandbytes was failing.

parent a4532c59
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
bitsandbytes/__main__.py
View file @ 26efb154
...@@ -30,7 +30,7 @@ print() ...@@ -30,7 +30,7 @@ print()
from . import COMPILED_WITH_CUDA, PACKAGE_GITHUB_URL 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 .cuda_setup.env_vars import to_be_ignored
from .utils import print_stderr from .utils import print_stderr
...@@ -47,7 +47,8 @@ print( ...@@ -47,7 +47,8 @@ print(
print_header("OTHER") print_header("OTHER")
print(f"{COMPILED_WITH_CUDA = }") 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("")
print_header("DEBUG INFO END") print_header("DEBUG INFO END")
print_header("") print_header("")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment