python - 没有名为 'uff' 的模块

Question

目标: 将tensorflow .pb模型转换为tensorrt

系统规范:乌本杜 18.04库达10.0TensorRT 5.1

甚至已安装sudo apt-get install uff-converter-tf

尝试在 Python 中导入 uff 时出现错误:

ModuleNotFoundError: No module named 'uff'

Answer 1

正在安装uff-converter-tf还不够，您需要安装Python UFF Wheel才能使用它。 Here您将找到完整的安装指南。

如果使用 Python 2.7:

pip2 install tensorrt-*-cp27-none-linux_x86_64.whl

如果使用 Python 3.x:

pip3 install tensorrt-*-cp3x-none-linux_x86_64.whl

您会发现这些轮子以及解压后生成的 TensorRT 二进制文件的其余部分 TensorRT-${version}.${os}.${arch}-gnu.${cuda}.${cudnn}.tar.gz 。

一如既往，最好安装在虚拟环境中。