cudnn安装

官网下载链接

安装CUDA

cudnn是应用很广泛的计算加速库

大多数教程是直接把include和lib中的文件覆盖了，需要重新添加软链接

复制include文件夹中的头文件，lib中的动态库文件在环境变量中指定路径，这种方法更加合理

tar -zvxf cudnn-***.tar.gz -C /your/path/to/cudnn
#解压到指定路径


vim ~/.bashrc

export PATH=/your/path/to/cuda/bin:$PATH
export LD_LIBRARY_PATH=/your/path/to/cuda/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/your/path/to/cudnn/lib:$LD_LIBRARY_PATH

source ~/.bashrc
#修改.bashrc文件


cd /your/path/to/cudnn/cuda/include

sudo cp *.h /usr/local/cuda/include/

sudo chmod a+r /usr/local/cuda/include/cudnn.h

查看cuda版本和cudnn版本

nvcc -V
#查看CUDA版本



cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
#查看cudnn版本