NVIDIA cuDNN - NVIDIA 文档

The NVIDIA CUDA® 深度神经网络 (cuDNN) 是一个 GPU 加速的深度神经网络基元库。cuDNN 为标准例程（如前向和后向卷积、注意力机制、矩阵乘法、池化和归一化）提供高度优化的实现。

发行说明

NVIDIA cuDNN 的当前状态、软件版本和已知问题。

安装指南

本文档提供有关如何安装 NVIDIA cuDNN 的分步说明。

库 API

此库是基于上下文的 API，可以轻松实现多线程以及与 CUDA 流的（可选）互操作性。此 API 列出了每个子库的数据类型和 API 函数。

开发者指南

本文档解释了如何使用 NVIDIA cuDNN 库。虽然 NVIDIA cuDNN 库 API 提供了每个函数的 API 文档，但《开发者指南》更非正式地讲述了 cuDNN 的关键功能以及如何使用它们。

故障排除

本文档描述了用于记录 cuDNN API 执行和错误信息的错误报告和 API 日志记录实用程序。它还有助于解答关于典型用例的最常见问题。

支持矩阵

本文档列出了最新 NVIDIA cuDNN 版本支持的操作系统、NVIDIA CUDA、CUDA 驱动程序和硬件版本。

软件许可

本文档包含 NVIDIA cuDNN 的具体许可条款和条件。接受本协议即表示您同意遵守适用于此处包含的特定产品的所有条款和条件。

归档

本文档提供对先前发布的 cuDNN 文档版本的访问权限。

cuDNN 库

单击描述目标平台的绿色按钮。只会显示受支持的平台。

cuDNN 前端

cuDNN 前端 (FE) API 是一个仅限 C++ 头文件的库，它封装了 cuDNN C 后端 API。FE 和后端 API 都是同一组功能的入口点，通常被称为“graph API”。

NVIDIA 开发者计划

加入 NVIDIA 开发者计划。

cuDNN 论坛

浏览 cuDNN 论坛。

cuDNN 开发者

访问最新的 NVIDIA cuDNN 公告、新闻、下载和培训。

技术博客

查找更多新闻和教程。