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

How to know which cuDNN version one should use?

我计划在 Linux 上使用 cuDNN:如何知道我需要哪个 cuDNN 版本?我应该始终使用最新的吗?

例如稍后选择正确的 CUDA 版本 the Nvidia driver version. I wonder if there is similar constrains for choosing the cuDNN (given that it may give some fancy 错误消息我希望在尝试之前知道)。

您应该使用您的应用程序和平台支持的最新版本的 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 的下载页面上找到。以前的版本可以从下载页面底部的 link 下载。

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

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

维基百科的 CUDA article has a good list of the various compute capability levels and corresponding GPUs (a similar list is available on Nvidia's compute capability 网页)。