2024-10-03 02:43:11 发布
网友
目标: 将tensorflow.pb模型转换为tensorrt
系统规格: 乌本杜18.04 Cuda 10.0 TensorRT 5.1
甚至安装了sudo apt-get install uff-converter-tf
sudo apt-get install uff-converter-tf
尝试在Python中导入uff时出错:
uff
ModuleNotFoundError: No module named 'uff'
安装uff-converter-tf是不够的,您需要安装Python UFF轮子才能使用它Here您将找到完整的安装指南
uff-converter-tf
如果使用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产生的
TensorRT-${version}.${os}.${arch}-gnu.${cuda}.${cudnn}.tar.gz
和往常一样,最好安装在虚拟环境中
安装
uff-converter-tf
是不够的,您需要安装Python UFF轮子才能使用它Here您将找到完整的安装指南如果使用Python 2.7:
如果使用Python 3.x:
您将找到这些控制盘和其他TensorRT二进制文件,它们是解包
TensorRT-${version}.${os}.${arch}-gnu.${cuda}.${cudnn}.tar.gz
产生的和往常一样,最好安装在虚拟环境中
相关问题 更多 >
编程相关推荐