是否有兼容 Cuda 9.0 和 cudnn 7.1 的 tensorflow 版本

Question

我有一台带有cuda 9.0 和cudnn 7.1 的机器。

我尝试在这台机器上使用tensorflow 1.7.0，但它不起作用，因为这个版本的tensorflow 是为cudnn 7.0 创建的 在我的 gpu 上启动训练时出现此错误：

Loaded runtime CuDNN library: 7102 (compatibility version 7100) but source was compiled with 7005 (compatibility version 7000).

是否有与我的cuda 和cudnn 版本兼容的tensorflow 版本？我还需要这个工作 tensorflow 版本是>=1.7.0。

我用谷歌搜索过这个问题，搜索了每个问题，但我从未得到cuda 和cudnn 的这些特定版本的答案。

Answer 1

tensorflow_gpu-1.9.0 应该可以做到这一点。下面链接的表格显示了 CUDA 和 cuDNN 与不同版本的 tensorflow 的兼容性。

https://www.tensorflow.org/install/install_sources#tested_source_configurations

Answer 2

好的，我好像错过了一些安装步骤。 通过安装tensorflow 的最新版本，在撰写本文时它是1.9.0，它确实可以在我的机器上运行。