是否有与Cuda 9.0和Cudnn 7.1兼容的Tensorflow版本

时间:2018-07-24 09:04:52

标签: tensorflow cudnn

我有一台装有cuda 9.0cudnn 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).

是否有与我的tensorflowcuda版本兼容的cudnn版本?我还需要这个工作的tensorflow版本为>=1.7.0

我已经用Google搜索了这个问题,搜索了每个问题,但对于这些特定版本的cudacudnn我一无所获。

2 个答案:

答案 0 :(得分:7)

tensorflow_gpu-1.9.0应该可以实现。下面的链接是一个表格,其中显示了不同版本的tensorflow的CUDA和cuDNN的兼容性。

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

答案 1 :(得分:1)

好吧,看来我错过了一些安装步骤。 通过安装最新版本的tensorflow(在撰写本文时为1.9.0),它可以在我的机器上正常工作。