【发布时间】:2019-03-04 08:27:59
【问题描述】:
在GPU上运行的tensorflow 1.11当前版本的requirements是
- CUDA® Toolkit —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的要求?
【问题讨论】:
-
我尝试使用 Cuda 9.0、cudnn 7.3 安装 TF 1.11 - 该设置适用于 TF 1.10,但 TF 1.11 挂在
tf.Session()上。不过我没有尝试任何其他组合。很高兴听到您是否成功。 -
面临同样令人困惑的问题。 @emilyfy 友好共享的库适用于 Ubuntu 和 AMD CPU,但不适用于带有 Intel 的 Windows 10。
标签: tensorflow cudnn