cudnn配置相比cuda简单了许多,首先需要下载,附上链接点击此处。安装时需要注册nvidia账户,可以使用微信或者QQ注册,选择自己对应的版本即可
2.选择下载对应版本的cudnn,解压,打开所在文件夹,执行下面的指令
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*
3.接着再从终端打开
sudo gedit ~/.bashrc
4.在文档尾部加上
export LD_LIBRARY_PATH = /usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
5.添加完路径后
source ~/.bashrc
6.到此为止CuDNN就安装好了,可以使用以下命令进行查看
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2