没有名为 '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
。
一如既往,最好安装在虚拟环境中。
目标: 将 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
。
一如既往,最好安装在虚拟环境中。