我有一台装有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).
是否有与我的tensorflow
和cuda
版本兼容的cudnn
版本?我还需要这个工作的tensorflow版本为>=1.7.0
。
我已经用Google搜索了这个问题,搜索了每个问题,但对于这些特定版本的cuda
和cudnn
我一无所获。
答案 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
),它可以在我的机器上正常工作。