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