我打算在Linux上使用cuDNN:如何知道我需要哪个cuDNN版本?我应该总是使用最新的?如何知道应该使用哪个cuDNN版本?
E.g.选择正确的CUDA版本depends on Nvidia驱动程序版本。我想知道是否有类似的选择cuDNN的限制(考虑到它可能会给我一些fancy错误消息,我想在尝试之前知道)。
我打算在Linux上使用cuDNN:如何知道我需要哪个cuDNN版本?我应该总是使用最新的?如何知道应该使用哪个cuDNN版本?
E.g.选择正确的CUDA版本depends on Nvidia驱动程序版本。我想知道是否有类似的选择cuDNN的限制(考虑到它可能会给我一些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的下载页面上。以前的版本可以从下载页面底部的链接下载。
除了CUDA版本的要求,你需要确保你的GPU具有计算能力足够高:
维基百科的文章CUDA有各种计算能力水平和相应的GPU好名单(类似的名单,以供Nvidia的compute capability网页)。
您是否在CUDA 6.5上尝试过cuDNN v3?如果使用较低版本的CUDA,潜在的问题是什么? – Jon
根据用户指南,cuDNN v5.1也适用于ARM平台的CUDA Toolkit 7.0。 –