NVIDIA CUDA 深度神经网络库- cuDNN

2021-02-02 16:49:17 浏览数 (1)

NVIDIA cuDNN

NVIDIA CUDA® 深度神经网络库 (cuDNN) 是经 GPU 加速的深度神经网络基元库。cuDNN 可大幅优化标准例程(例如用于前向传播和反向传播的卷积层、池化层、归一化层和激活层)的实施。

世界各地的深度学习研究人员和框架开发者都依赖 cuDNN 实现高性能 GPU 加速。借助 cuDNN,研究人员和开发者可以专注于训练神经网络及开发软件应用,而不必花时间进行低层级的 GPU 性能调整。cuDNN 可加速广泛应用的深度学习框架,包括 Caffe2、Chainer、Keras、MATLAB、MxNet、PyTorch 和 TensorFlow](https://www.tensorflow.org/)。如需获取经 NVIDIA 优化且已在框架中集成 cuDNN 的深度学习框架容器,请访问 NVIDIA GPU CLOUD 了解详情并开始使用。

CUDA 工具包 11.1 下载地址为:https://developer.nvidia.com/zh-cn/cuda-downloads

![CUDA 工具包 11.1 Windows版本下载

相关资料

  • 面向开发者的 NVIDIA AI 深度学习平台
  • CUDA 工具包 11.1 下载地址

0 人点赞