解决 nvcc -V 显示的cuda版本不正确

假设你刚安装了新的CUDA版本,以CUDA-10.1为例,安装完成后用nvcc -V查看,输出的仍然是之前的老版本CUDA。

使用

~$ which nvcc
/usr/bin/nvcc

 查看系统默认调用的nvcc命令的位置,

~$ cat /usr/bin/nvcc
exec /usr/lib/nvidia-cuda-toolkit/bin/nvcc

 这是指向了系统默认的nvidia-cuda-toolkit位置,将其更改为刚安装的cuda:

~$ sudo vi /usr/bin/nvcc
exec /usr/local/cuda/bin/nvcc

 保存更改后,nvcc -V命令得到CUDA-10.1版本。

nvcc -V输出版本不正确会导致较多cuda代码编译的问题,比如detectron2中的自定义算子编译步骤。

按此方法更改后可正确编译。 

 

posted @ 2021-04-27 09:55  Gaaray  阅读(5722)  评论(1编辑  收藏  举报