def main()

in bitsandbytes/diagnostics/main.py [0:0]

• 34 lines of code
• 5 McCabe index (conditional complexity)

def main():
    print_header("")
    print_header("BUG REPORT INFORMATION")
    print_header("")

    print_header("OTHER")
    cuda_specs = get_cuda_specs()
    print("CUDA specs:", cuda_specs)
    if not torch.cuda.is_available():
        print("Torch says CUDA is not available. Possible reasons:")
        print("1. CUDA driver not installed")
        print("2. CUDA not installed")
        print("3. You have multiple conflicting CUDA libraries")
    if cuda_specs:
        print_cuda_diagnostics(cuda_specs)
    print_cuda_runtime_diagnostics()
    print_header("")
    print_header("DEBUG INFO END")
    print_header("")
    print("Checking that the library is importable and CUDA is callable...")
    try:
        sanity_check()
        print("SUCCESS!")
        print("Installation was successful!")
        return
    except ImportError:
        print(
            f"WARNING: {__package__} is currently running as CPU-only!\n"
            "Therefore, 8-bit optimizers and GPU quantization are unavailable.\n\n"
            f"If you think that this is so erroneously,\nplease report an issue!",
        )
    except Exception:
        traceback.print_exc()
    print_dedented(
        f"""