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

Question

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

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

Answer 1

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

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 的下载页面上找到。可以从下载页面底部的链接下载以前的版本。

Answer 2

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

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

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