CUDA版本:11.1
第一步:选择cudnn版本
下载链接:https://developer.nvidia.com/rdp/cudnn-archive
注:根据CUDA版本选择对应的cudnn,本人CUDA版本为11.1,所以这里选择cuDNN v8.0.5
第二步:下载文件
注:如果系统是Ubuntu18.04,最好下载截图中与Ubuntu18.04 x86_64相对应的三个文件(Runtime,Developer,Code Samples and User Guide),而不是cuDNN Library for Linux [x86_64]
第三步:开始安装
进入下载后上述三个文件所在的文件夹,打开终端输入(前两行顺序不能错,否则无法安装成功)
sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.1_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.1_amd64.deb
第四步:修改文件位置
把.h文件复制到/usr/local/cuda/include文件夹下面,并修改权限
sudo cp /usr/include/cudnn.h /usr/local/cuda/include
sudo chmod a+x /usr/local/cuda/include/cudnn.h
第五步:验证cuDNN是否安装成功
为了验证 cuDNN 已经安装并正确运行,需要编译位于 /usr/src/cudnn_samples_v8 目录下的 mnistCUDNN 样例:
·将cuDNN示例复制到可写路径
cp -r /usr/src/cudnn_samples_v8/ $HOME
· 进入到可写路径
cd $HOME/cudnn_samples_v8/mnistCUDNN
·编译mnistCUDNN示例
make clean && make
·运行mnistCUDNN示例
./mnistCUDNN
运行后出现Test passed!,则说明安装成功