没有名为 'uff' 的模块

No module named 'uff'

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

系统规格: 乌本都 18.04 库达 10.0 TensorRT 5.1

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

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

ModuleNotFoundError: No module named 'uff'

安装 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

您会发现这些 wheels 与解包后产生的其余 TensorRT 二进制文件 TensorRT-${version}.${os}.${arch}-gnu.${cuda}.${cudnn}.tar.gz

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