Bitsandbytes 文件
故障排除
加入 Hugging Face 社群
並獲得增強的文件體驗
開始使用
問題排查
無可用核心映象 (No kernel image available)
當 bitsandbytes 載入的 CUDA 版本不被您的 GPU 支援,或者您的 PyTorch CUDA 版本不匹配時,會出現此問題。
要解決此問題,您需要除錯 $LD_LIBRARY_PATH
、$CUDA_HOME
以及 $PATH
。您可以透過 echo $PATH
列印這些變數。您應該查詢指向不同 CUDA 版本的多個路徑。這可能包括您 anaconda 路徑中的版本,例如 $HOME/anaconda3/lib
。您可以透過 ls -l $HOME/anaconda3/lib/*cuda*
或等效路徑來檢查這些版本。檢視這些路徑中檔案的 CUDA 版本。它是否與 nvidia-smi
的輸出匹配?
如果您想碰碰運氣,也可以嘗試從原始碼編譯該庫。如果您的 PATH 變數中有多個 CUDA 版本,這仍然可能會有問題。因此,建議在繼續編譯之前解決路徑衝突。
fatbinwrap
如果 C++ 庫和 CUDA 部分的 CUDA 版本不匹配,則會發生此錯誤。請確保您的 $PATH
和 $LD_LIBRARY_PATH
變數中有正確的 CUDA。在 conda 基礎環境中,您可以在以下位置找到該庫:
ls $CONDA_PREFIX/lib/*cudart*
請確保將此路徑附加到 LD_LIBRARY_PATH
中,以便 bnb 可以找到 CUDA 執行時環境庫 (cudart)。
如果這不能解決問題,請接下來嘗試從原始碼編譯。
如果這仍然不起作用,請提交一個 issue,並貼上您呼叫 make
時列印的環境資訊以及執行 bnb 時的相關錯誤。