Commit 0acab1c6 authored by Titus von Koeller's avatar Titus von Koeller
Browse files

case 3,4a/b: no lib but cuda validated

parent 71a9ce41
...@@ -135,16 +135,18 @@ class ErrorHandlerMockBNBNativeLibrary(BNBNativeLibrary): ...@@ -135,16 +135,18 @@ class ErrorHandlerMockBNBNativeLibrary(BNBNativeLibrary):
elif no_cuda_lib_found: elif no_cuda_lib_found:
version_list_str = "\n - " + "\n - ".join(available_versions) if available_versions else "NONE" version_list_str = "\n - " + "\n - ".join(available_versions) if available_versions else "NONE"
analysis = ( analysis = (
f"\n🚨 CUDA VERSION MISMATCH 🚨\n" (
f"Requested CUDA version: {requested_version}\n" f"\n🚨 CUDA VERSION MISMATCH 🚨\n"
f"Detected PyTorch CUDA version: {user_cuda_version}\n" f"Requested CUDA version: {requested_version}\n"
f"Available pre-compiled versions: {version_list_str}\n\n" f"Detected PyTorch CUDA version: {user_cuda_version}\n"
"This means:\n" f"Available pre-compiled versions: {version_list_str}\n\n"
"The version you're trying to use is NOT distributed with this package\n\n" "This means:\n"
"The version you're trying to use is NOT distributed with this package\n\n"
)
if available_versions if available_versions
else "1. You're not using the package but checked-out the source code\n" else "\n🚨 Forgot to compile the bitsandbytes library? 🚨\n"
"2. You MUST compile from source for this specific CUDA version\n" "1. You're not using the package but checked-out the source code\n"
"3. The installation will NOT work until you compile or choose a CUDA supported version via export BNB_CUDA_VERSION=<version>\n\n" "2. You MUST compile from source\n\n"
) )
base_msg = "Attempted to use bitsandbytes native library functionality but it's not available.\n\n" base_msg = "Attempted to use bitsandbytes native library functionality but it's not available.\n\n"
......
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