解决 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中的自定义算子编译步骤。
按此方法更改后可正确编译。