Bitsandbytes 文件

故障排除

Hugging Face's logo
加入 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 時的相關錯誤。

< > 在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.