cuDNN神经网络加速库的配置

1.简介

官网介绍:

NVIDIA CUDA® 深度神经网络库 (cuDNN) 是一个 GPU 加速的深度神经网络基元库,能够以高度优化的方式实现标准例程(如前向和反向卷积、池化层、归一化和激活层)。

全球的深度学习研究人员和框架开发者都依赖 cuDNN 来实现高性能 GPU 加速。借助 cuDNN,研究人员和开发者可以专注于训练神经网络及开发软件应用,而不必花时间进行低层级的 GPU 性能调整。cuDNN 可加速广泛应用的深度学习框架,包括 Caffe2、Chainer、Keras、MATLAB、MxNet、PaddlePaddle、PyTorch 和 TensorFlow。

自己理解:

使用cuDNN可以对深度学习训练进行加速

2.配置过程

(1)到cuDNN下载网址,结合自身CUDA版本进行下载.因为我的CUDA版本为10.1,所以我下载这个!!!
cuDNN神经网络加速库的配置_第1张图片
(2)根据自己的电脑系统进行选择cuDNN库
cuDNN神经网络加速库的配置_第2张图片
(3)没有NVIDIA账号的小伙伴创建一个,登录之后进行下载.
cuDNN神经网络加速库的配置_第3张图片
(4)对下载好的压缩包进行解压,复制这三个文件夹到之前安装CUDA的CUDA1文件夹下
没有安装好CUDA的小伙伴可以看看我之前这篇博文:
https://blog.csdn.net/Elon15/article/details/125678410?spm=1001.2014.3001.5502
cuDNN神经网络加速库的配置_第4张图片

cuDNN神经网络加速库的配置_第5张图片
(5)复制完成,表示cuDNN神经网络加速库配置完成!!!
cuDNN神经网络加速库的配置_第6张图片

附录:

参考了B站up主:有幸遇见的上上

你可能感兴趣的:(PyTorch,神经网络,深度学习,人工智能)