解决 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中的自定义算子编译步骤。

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

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM