1.下载文件
首先进入cuDNN官网首页,通过邮箱注册后登陆,进入Download页面。
根据操作系统和系统中安装的cuda版本选择合适的cuDNN版本。download界面如下所示:
本文选择cuDNN Library for Linux进行下载。如果在windows系统下载此文件,选择在迅雷中下载。下载后在ubuntu的home中解压。解压后在home路径下执行以下指令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
执行上述指令后就安装完成了,输入下面指令验证是否正确安装:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如显示如下类似内容,显示cuDNN的版本信息,证明已经正确安装。