Ubuntu cudnn安装

Ubuntu cudnn安装

确定成功安装cuda

nvcc -V查看的版本

我安装cuda版本11.3,因此我安装的cudnn也应该是11.3

cudnn官网找与cuda对应版本的cudnn

cudnn官网 https://developer.nvidia.com/rdp/cudnn-archive
下载tar.xz文件,通过xftp软件传到服务器

解压缩

// 朋友们注意自己下载文件的名称
tar -xvf cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xz

复制部分文件

接下来将部分文件进行复制cuda11.3里面

// 朋友们注意自己下载文件的名称
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*

注意一下:cudnn8.0以上会将版本信息更新到cudnn_version.h文件中,也需要将这个文件复制,否则验证的时候会没有反应

// 朋友们注意自己下载文件的名称
sudo cp cuda/include/cudnn_version.h    /usr/local/cuda/include/

验证成功

// 朋友们注意自己下载文件的名称
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

出现如下信息即成功