Ubuntu cudnn安装

Posted 2023-03-31 竹子2021

Ubuntu cudnn安装,先安装好cuda

• Ubuntu cudnn安装
• • 确定成功安装cuda
  • cudnn官网找与cuda对应版本的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

出现如下信息即成功

ubuntu 下安装 cudnn

安装cudnn前后，GPU跑一个算法的速度分别是139ms和26ms ！

 

1. 在以下网址选择  cuDNN v5.1 Library for Linux 下载

https://developer.nvidia.com/rdp/cudnn-download

 

2.下载得到一个 cudnn-8.0-linux-x64-v5.1.tgz 包，解压之

得到 cuda 文件夹， 下含两个子文件夹 include , lib64

 

3. 把 include 和 lib64 文件夹内容分别复制到

/usr/local/cuda/include

/usr/local/cuda/lib64