o oiR @sdddlZddlZddlZddlmZddlmZddlmZm Z ddl m Z m Z ddZ dd ZdS) N)PACKAGE_GITHUB_URL)get_cuda_specs)print_cuda_diagnosticsprint_cuda_runtime_diagnostics)print_dedented print_headerc Csddlm}|durtdddlm}tjtdd }tdd }|j }||g}||}| }| ||j }||ksQJdS)Nr)liba Couldn't load the bitsandbytes library, likely due to missing binaries. Please ensure bitsandbytes is properly installed. For source installations, compile the binaries with `cmake -DCOMPUTE_BACKEND=cuda -S .`. See the documentation for more details if needed. Trying a simple check anyway, but this will likely fail... )Adam )bitsandbytes.cextensionrrbitsandbytes.optimr torchnn Parameterrandcudadatasumitembackwardstep) rr pap1adamoutlossp2rQ/home/ubuntu/.local/lib/python3.10/site-packages/bitsandbytes/diagnostics/main.py sanity_checks   r cCstdtdtdtdt}td|tjs-tdtdtdtd|r3t|ttdtd tdtd zttd td WdSt yetd t dYn t yqt YnwtdtdtddS)NzBUG REPORT INFORMATIONOTHERz CUDA specs:z3Torch says CUDA is not available. Possible reasons:z1. CUDA driver not installedz2. CUDA not installedz/3. You have multiple conflicting CUDA librarieszDEBUG INFO ENDz?Checking that the library is importable and CUDA is callable...zSUCCESS!zInstallation was successful!z WARNING: z is currently running as CPU-only! Therefore, 8-bit optimizers and GPU quantization are unavailable. If you think that this is so erroneously, please report an issue!zm Above we output some debug information. Please provide this info when creating an issue via zy/issues/new/choose WARNING: Please be sure to sanitize sensitive info from the output before posting it. )rrprintr r is_availablerrr ImportError __package__ Exception traceback print_excrrsysexit) cuda_specsrrrmain-sH      r.)r+r)r bitsandbytes.constsrbitsandbytes.cuda_specsrbitsandbytes.diagnostics.cudarrbitsandbytes.diagnostics.utilsrrr r.rrrrs