查看CUDA版本:
cat /usr/local/cuda/version.txt
查看 CUDNN 版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
版本不一致解决方法:
1.使用sudo apt-get autoremove nvidia-cuda-toolkit 卸载7.5版本
sudo apt-get autoremove nvidia-cuda-toolkit
2.查看/usr/local/cuda/bin下是否有nvcc可执行程序,如果没有说明cuda没有正常安装,需要重新安装,如果有,进入下一步
cd /usr/local/cuda/bin
ls
3.添加环境变量,在用户目录下输入
ls -a ,找到.bashrc文件,
输入sudo gedit ./.bashrc 打开~/.bashrc ,添加环境变量export PATH="/usr/local/cuda/bin:$PATH"
ls -a
sudo gedit ./.bashrc