Tensorflow 1.11需要CUDA 9.0的CuDNN 7.2,但没有这样的库

时间:2018-09-28 10:44:56

标签: tensorflow cudnn

要在GPU上运行的当前版本的Tensorflow 1.11的requirements

  
      
  • CUDA®工具包— TensorFlow支持CUDA 9.0。
  •   
  • cuDNN SDK(> = 7.2)
  •   

但是CuDNN downlad page仅列出

  

下载cuDNN v7.2.1(2018年8月7日),适用于CUDA 9.2

鉴于CuDNN为CUDA工具包的次要版本提供了不同的二进制文件(例如,CuDNN 7.1.3具有一个用于CUDA 9.1的二进制文件,另一个具有用于CUDA 9.0的二进制文件),我想此CuDNN 7.2的二进制文件与CUDA 9.0不兼容。

是文档错误吗?如果没有,如何满足TF 1.11的要求?

3 个答案:

答案 0 :(得分:4)

我通过修改公共库中的地址找到了以下答案:https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.2.1/prod/9.0_20180806/cudnn-9.0-windows10-x64-v7.2.1.38

正如@emilyfy所建议的,也可以获取其他托管但未发布的版本和操作系统的地址。

答案 1 :(得分:3)

改为转到此页面。 https://developer.nvidia.com/rdp/cudnn-download

具有指向的链接 Download cuDNN v7.3.0 (Sept 19, 2018), for CUDA 9.0

答案 2 :(得分:2)

CUDA 9.0的

cuDNN v7.2.1曾经存在,但是现在有了v7.3.0,它不再存在于归档中。我在另一台PC上构建的模型也遇到了同样的问题。幸运的是,我没有删除安装程序。我将与他们分享(仅适用于Linux的deb安装程序)here