Deformable-detr在运行make.sh文件时,报NotImplementedError: Cuda is not availabel

NotImplementedError: Cuda is not availabel

当你运行下面命令报错NotImplementedError: Cuda is not availabel

sh make.sh

根据我的经验,列出几条可以排查的思路。

  1. 使用nvcc -V查看一下在当前虚拟环境中使用的cuda版本。这个版本是cuda的编译版本;而在虚拟环境中,在pytorch安装的cuda是运行版本,编译时的版本不能高于运行版本,否则会报版本mismatch(这都是踩过的坑啊!!!)。
  2. 使用echo $CUDA_HOME命令查看CUDA_HOME的配置路径,这就相当于Windows命令下,cuda的环境变量配置。在服务器cuda可用的情况下,报Cuda is not available错误,就是CUDA_HOME没有配置好造成的。如果键入命令echo $CUDA_HOME没有输出的话,就是没有配置。可以参考下面的命令!
export LD_LIBRARY_PATH=/home/chenxu/cuda-11.8/lib64$LD_LIBRARY_PATH
export CUDA_HOME=/home/chenxu/cuda-11.8
export PATH=/home/chenxu/cuda-11.8/bin:$PATH
  1. 我之所以报错是因为服务器的cuda驱动没有安装好,驱动安装好了之后因为版本太高又报了上述提到的版本mismatch的问题。

有什么问题评论区留言,时间问题不细写了。这是我遇到的棘手的错误,好多天才solve。希望能给大家提供了solutions。Keep Moving!

猜你喜欢

转载自blog.csdn.net/weixin_44768070/article/details/132018163