首先,我们知道NVIDIA官网更新了所有的cudnn安装包,之前都是tar.gz格式,我们是通过解压文件在安装。但是现在如果是通过Deb包安装,则查看版本是不一样的,检验是否安装成功命令也不太一样。
1.如果是tar.gz格式的安装查看cuda的版本:
cat /usr/local/cuda/version.txt
查看cudnn 版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
2.如果是Deb安装版本
比如我们安装的是7.0.5版本cudnn,则要下载三个文件:
cuDNN v7.0.5 Runtime Library for Ubuntu16.04 (Deb)
cuDNN v7.0.5 Develpoper Library for Ubuntu16.04 (Deb)
cuDNN v7.0.5 Code Samples and User Guide for Ubuntu16.04 (Deb)
cudnn下载
此时,
查看cuda的版本:
cat /usr/local/cuda/version.txt
查看cudnn 版本:
使用原来的cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
命令会提示找不到cudnn.h文件,原因是
新版cudnn Deb安装版本 找不到cudnn.h,他不会有这个文件。
应该使用命令:
dpkg -l | grep cudnn
比如我的是cudnn7,有三个文件。
检验cudnn是否安装成功
找到cudnn_samplesv7文件夹,
如图,我的检验方法是:
cd mnistCUDNN
运行里面的mnistCUDNN脚本
./mnistCUDNN
显示出 Test passed!说明安装成功。