Tensorflow可以在一个python文件中找到正确的cudnn,但在另一个python文件中失败

时间:2018-12-30 09:51:35

标签: tensorflow gpu nvidia cudnn

我正在尝试使用tensorflow gpu版本来训练和测试我的深度学习模型。但是问题来了。当我在一个python文件中训练模型时,一切进展顺利。 Tensorflow-gpu可以正确使用。然后,我将模型保存为经过预先训练的grapg.pb格式,并尝试在另一个python文件中重用它。

然后我收到以下错误消息。

E tensorflow/stream_executor/cuda/cuda_dnn.cc:363] Loaded runtime CuDNN 
library: 7.1.4 but source was compiled with: 7.2.1.  CuDNN library major 
and minor version needs to match or have higher minor version in case of 
CuDNN 7.0 or later version. If using a binary install, upgrade your CuDNN 
library.  If building from sources, make sure the library loaded at runtime 
is compatible with the version specified during compile configuration.

我检查了我的cudnn版本,实际上它是版本7.4.2。我还检查了其中的环境路径设置/cuda/v9.0/bincuda/v9.0/lib/x64/cuda/v9.0/include

那为什么会这样呢?我该如何解决呢?

-

  

cuda:v9.0   cudnn:7.4.2(我认为,我手动复制了这些cudnn文件)   窗户10   python:3.5

1 个答案:

答案 0 :(得分:0)

如果您通过各种方式安装了多个 CuDNN,例如 anaconda 模块和 windows 安装,您需要删除旧版本,以便您的代码检测到最新版本并重新安装 tensorflow-gpu
您可以按照this指南进行基于操作系统的安装。