1、cuDNN简介
cuDNN的全称为NVIDIA CUDA® Deep Neural Network library,是NVIDIA专门针对深度神经网络中的基础操作而设计并基于GPU的加速库。
cuDNN为深度神经网络中的标准流程提供了高度优化的实现方式,例如:convolution、pooling、normalization以及activation layers的前向以及后向过程。
安装CUDA的时候并没有安装cuDNN库,当开发者们需要用到深度学习GPU加速时才安装cuDNN库。
2、cuDNN与CUDA的版本对应关系
https://developer.nvidia.com/rdp/cudnn-archive
3、cuDNN下载地址
https://developer.nvidia.com/cudnn
4、cuDNN安装
下载好cuDNN的压缩包后进行解压,得到如下三个文件:
bin
include
lib
打开\NVIDIA Computing Toolkit\CUDA\v11.8
,将cuDNN对应bin、lib、include三个文件与CUDA对应的bin、lib、include进行合并。
读者推荐
要想彻底的掌握pytorch的方方面面,推荐大家关注《PyTorch面试精华》