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

Question

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

E.g。选择正确的CUDA版本depends on Nvidia驱动程序版本。我想知道是否存在类似的选择cuDNN的约束（假设它稍后可能会给出一些fancy错误消息，我想在尝试之前知道）。

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

Answer 2

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

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

Wikipedia＆＃39; CUDA文章列出了各种计算能力水平和相应的GPU（Nvidia的compute capability网页上提供了类似的列表）。