我正在尝试使用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/bin
,cuda/v9.0/lib/x64
,/cuda/v9.0/include
。
那为什么会这样呢?我该如何解决呢?
-
cuda:
v9.0
cudnn:7.4.2
(我认为,我手动复制了这些cudnn文件) 窗户10 python:3.5
答案 0 :(得分:0)
如果您通过各种方式安装了多个 CuDNN
,例如 anaconda 模块和 windows 安装,您需要删除旧版本,以便您的代码检测到最新版本并重新安装 tensorflow-gpu
。
您可以按照this指南进行基于操作系统的安装。