cuDNN安装

可keke

已于 2024-07-25 19:46:39 修改

阅读量549

点赞数 5

CC 4.0 BY-SA版权

分类专栏：环境软件安装配置相关文章标签：深度学习

于 2024-07-21 15:45:02 首次发布

本文链接：https://blog.csdn.net/qq_44722189/article/details/140588960

环境软件安装配置相关专栏收录该内容

20 篇文章

订阅专栏

文章目录

需要已经安装好了Nvidia驱动及CUDA。这里主要记录一下使用tar包安装cuDNN的方式，官方文档中还有其他的安装方式，可以参考最新官方文档或历史版本官方文档。

下载

在这里可以下载最新版本的cuDNN，但一般我们是要去历史版本中寻找对应目前使用中的CUDA的版本下载。根据自己系统版本及处理器架构选择对应的安装包，我使用的是第二个Local Installer for Linux x84_64(Tar)。

根据系统选择对应的安装包，我使用的是第二个

安装

下载完tar包后，在tar包所在的目录下开始进行安装。
官方文档中的安装步骤：

在这里插入图片描述

$ tar -xvf cudnn-linux-$arch-8.x.x.x_cudaX.Y-archive.tar.xz
$ sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
$ sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

严格来说，CUDNN是个库，不需要安装，只需要把其文件复制到CUDA文件夹下即可使用。要把CUDNN中的bin，include，lib都复制到CUDA的对应目录下。注意，CUDA的bin文件夹应该被添加进环境变量。用默认设置安装CUDA时一般该文件夹会自动被加入环境变量。

验证

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

打印出类似如下内容说明安装成功：

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

Reference

https://blog.csdn.net/zxdd2018/article/details/127705627
https://zhouyifan.net/2021/08/19/20210813-PytorchInstall/