如何知道应该使用哪个 cuDNN 版本？

Posted 2023-04-15

【中文标题】如何知道应该使用哪个 cuDNN 版本？

【英文标题】：How to know which cuDNN version one should use?

【发布时间】：2016-02-19 00:23:07

【问题描述】：

我打算在 Linux 上使用 cuDNN：如何知道我需要哪个 cuDNN 版本？我应该始终使用最新的吗？

例如选择正确的 CUDA 版本depends on Nvidia 驱动程序版本。我想知道选择 cuDNN 是否有类似的限制（考虑到它可能会在稍后给出一些 fancy 错误消息，我希望在尝试之前知道）。

【参考方案1】：

您应该使用您的应用程序和平台支持的最新版本的 cuDNN，因为这将有最多的错误修复和增强。是的，cuDNN 版本取决于特定的 cuda 版本。这在download page 中有详细说明。 （您必须是注册开发者才能访问该页面。）

cuDNN v5.1 has different versions for CUDA 7.5 and CUDA 8.0
cuDNN v5   has different versions for CUDA 7.5 and CUDA 8.0
cuDNN v4 and v3 both require CUDA 7.0
cuDNN v2 and v1 both require CUDA 6.5
All cuDNN versions require compute capability >= 3.0 devices

同样，以上所有信息都可以在 cuDNN 的下载页面上找到。以前的版本可以从下载页面底部的链接下载。

【讨论】：

您是否在 CUDA 6.5 上尝试过 cuDNN v3？如果使用较低版本的 CUDA，可能会出现什么问题？

根据用户指南，cuDNN v5.1 也适用于 ARM 平台的 CUDA Toolkit 7.0。

【参考方案2】：

除了 CUDA 版本要求，您还需要确保您的 GPU 具有足够高的计算能力：

从 1.0 到 5.1 的所有 cuDNN 版本都需要 3.0 或更高的计算能力。

Wikipedia 的 CUDA 文章很好地列出了各种计算能力级别和相应的 GPU（类似的列表可在 Nvidia 的 compute capability 网页上找到）。

【讨论】：

为了完整起见，我已将显着计算能力信息编辑到 @RobertCrovella 答案中

我们可以在哪里查看 CuDNN v4、v5 的信息？

@mrgloom，我已经更新了答案以反映新版本（它们都需要 3.0），但您可以查看未来版本的用户指南here（您需要一个帐户） .

有谁知道为什么 Nvidia 要求用户在他们的网站上注册才能访问用户指南和下载？