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 上更新