时间戳

首页 > 解决方案 > Tensorflow 1.11 需要 CUDA 9.0 的 CuDNN 7.2，但是没有这样的库

tensorflow - Tensorflow 1.11 需要 CUDA 9.0 的 CuDNN 7.2，但是没有这样的库

问题描述

当前版本的 tensorflow 1.11 在 GPU 上运行的要求是

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

然而，CuDNN 下载页面仅列出

下载 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 的要求？

标签： tensorflowcudnn

解决方案

我通过修改公开库中的地址找到了以下答案：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 建议的那样，还可以获得其他托管​​但未发布的版本和操作系统的地址。

推荐阅读