安装CUDNN 6.0【配 cuda 8.0】

1. 查看 cuda 和 cudnn 版本

命令: 

cat /usr/local/cuda/version.txt

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

2.  安装

从官网(https://developer.nvidia.com/rdp/cudnn-download)上下载需要的安装包:cudnn-8.0-linux-x64-v6.0.tgz 

解压处理:tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz

解压后得到一个cuda文件夹:

进入cuda文件夹,里面有:一个include文件夹,和一个lib64文件夹,复制一些东西到我们之前安装cuda的路径下(/usr/local/cuda/)

命令:

$cd cuda  
$sudo cp lib64/lib* /usr/local/cuda/lib64/  
$sudo cp include/cudnn.h /usr/local/cuda/include/

3. 更新cuDNN库文件的软连接

命令:

$ sudo chmod +r libcudnn.so.6.0.21
$ sudo ln -sf libcudnn.so.6.0.21  libcudnn.so.6
$ sudo ln -sf libcudnn.so.6 libcudnn.so
$ sudo ldconfig

4. 检验

命令:cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

5. 出现结果,成功

猜你喜欢

转载自blog.csdn.net/One1994/article/details/81910337