欢迎大家关注笔者，你的关注是我持续更博的最大动力

原创文章，转载告知，盗版必究

---

先说明一下我自己的环境：

• Ubuntu18.04
• cuda10.2
• cudnn7.6.5
• TensorRT-7.0.0.11
• uff0.6.5

---

1 错误原因分析

在导入uff库包时，报错如下：

这个错误时由于：GraphDef在TensorFlow1.x中是支持的，在TensorFlow2.x是不支持的，而我安装的tensorflow版本恰好是：tensorflow-gpu=2.2，因此导致这错误

2 错误解决方式

2.1 错误解决方式1（不推荐）

把tensorflow的版本降到1.x版本，如下：

pip install tensorflow-gpu==1.4.0

之所以不推荐这种方法，是因为Tensorflow1.x版本太老，用 的也比较少，而且不一定满足你现在的项目需求

2.2 错误解决方式2（推荐）

1、查看自己的uff库包的安装存放路径

2、修改conversion_helpers.py文件

conversion_helpers.py的路径位置

/home/shl/anaconda3/envs/TensorRT/lib/python3.7/site-packages/uff/converters/tensorflow/conversion_helpers.py

然后把conversion_helpers.py文件第18行的：from tensorflow import GraphDef 修改为from tensorflow.compat.v1 import GraphDef

3、在次查看uff能否导入成功

成功导入！！！

---

---

---

---

♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠ ⊕ ♠