当CUDA9安装完成后,需要进行检测:
1) cat /usr/local/cuda/version.txt #这种方法不太准,得到9.0.176
2) nvcc --version #这种方法准。如果nvcc没有安装,则sudo apt install nvidia-cuda-toolkit。检查出来是7.5.17
3)通过代码来查:
cudaDriverGetVersion(&driver_version); #获取cuda版本。得到9.0
cudaRuntimeGetVersion(&runtime_version); #获取cuda运行时版本。得到7.5
这种方法,例子是https://blog.csdn.net/langb2014/article/details/51348501
还有一个中文的例子https://www.2cto.com/kf/201708/666635.html
其中,CMakeLists.txt的写法,例子是https://blog.csdn.net/jz4626828/article/details/79878945
4)检测显卡型号及NVIDIA驱动版本
查看GPU型号 lspci | grep -i nvidia 得到M4000
查看NVIDIA驱动版本 sudo dpkg --list | grep nvidia-* 得到384.130
根据不同版本cuda对应的NVIDIA驱动版本(不同版本cuda对应的NVIDIA驱动版本 - LearnFromNow - 博客园),这个驱动应该对应cuda9.0
5)进一步查是否环境变量的问题,参见 Ubuntu16.04下安装多版本cuda和cudnn - tunhuzhuang1836的博客 - CSDN博客
在~/.