DL | TensorRT将Tensorflow模型转换为uff格式 报错Unable to locate package uff-converter-tf
前情概要：尝试用Nvidia的tensorRT加速模型，生成tf的pb模型后，用uff的python工具包进一步转换模型时，出现错误。
实验环境：TensorRT 5.0+CUDA10.0的nvidia-docker镜像环境
实验目的：尝试将tensorflow的pb模型解析为uff格式
实验途径：依赖于uff工具包，import uff成功
问题描述：
使用uff的api from_tensorflow_frozen_model 对pb模型进行转换。
出错信息如下：
AttributeError: module 'uff' has no attribute 'from_tensorflow_frozen_model'
问题分析：
首先判断是uff可能安装的不正确，尝试重新安装uff包。根据官网教程输入指令：apt-get install uff-converter-tf，再次报错，报错信息如下：
E: Unable to locate package uff-converter-tf
继续解决这个问题，发现资料很少，仔细研读官网教程（https://docs.nvidia.com/deeplearning/sdk/tensorrt-install-guide/index.html#installing-debian）发现Uff工具包已经打包在TRT的安装包里了，所以推断需要重新安装TRT。于是重新下载目标版本的TRT，注意匹配CUDA版本，我下载的是deb格式，使用指令本地解压：
dpkg -i nv-tensorrt-repo-ubuntu1x04-cudax.x-trt5.1.x.x-ga-yyyymmdd_1-1_amd64.deb
然后更新apt的源并安装TRT：
apt-get update && apt-get install tensorrt
安装python接口：
apt-get install python3-libnvinfer-dev
然后再进行uff包的安装就可以了！
apt-get install uff-converter-tf
这里的坑是我在之前直接apt-get install失败后，有些困惑，网上资料也很少。其实需要仔细的读现有的官方文档，uff的这个工具包已经包含在TRT里，当然也可以尝试找到镜像里TRT的安装路径，可能可以找到另一种解法。
预祝大家转换顺利！
DL | TensorRT将Tensorflow模型转换为uff格式 报错Unable to locate package uff-converter-tf相关推荐
- 将faster RCNN从github上下载的frozen模型转换为tflite时候报错
转换代码如下: tflite_convert --graph_def_file=/home/sir/project/models/faster/frozen_inference_graph.pb ...
- python response.json()报错_Django JsonResponse json格式报错 解决Django响应JsonResponse返回json格式数据报错问题...
想了解解决Django响应JsonResponse返回json格式数据报错问题的相关内容吗,彭世瑜在本文为您仔细讲解Django JsonResponse json格式报错的相关知识和一些Code实例 ...
- 解决vue项目格式报错问题_@jie
vue项目格式报错问题: 简述问题:写项目过程中格式一直报错,如下图: 第一步:在项目根目录下创建名为vue.config.js文件,如下图: 第二步:js文件中的内容,如下图: 代码如下: modu ...
- 将tensorflow模型转换为uff模型
将ckpt转换为pb模型 https://blog.csdn.net/guyuealian/article/details/82218092 将pb模型转换为uff模型 convert-to-uff ...
- 【TensorRT】PyTorch模型转换为ONNX及TensorRT模型
文章目录 1. PyTorch模型转TensorRT模型流程 2. PyTorch模型转ONNX模型 3. ONNX模型转TensorRT模型 3.1 TensorRT安装 3.2 将ONNX模型转换 ...
- 3DMAX模型转换为gltf格式总结与问题汇总
众所周知,cesium平台所支持的三维模型格式为3dtiles和gltf,这两种格式都是开源的.对于建的的精模如果数据量不是很大的话(单个文件在2G以内),可以使用gltf格式,但是在模型转换为glt ...
- 3dmax模型转换为*.FLT格式的建模要点
要想在3dmax中建立模型,并将3dmax模型转换为OpenFlight模型并且能够使用,在Max中建模时我们必须注意以下几点: 1.建立的模型要尽量采用比较简易的Box来表现,不能使用NURBS.p ...
- 将FBX模型转换为glb格式
这里提供一种最推荐的glTF转换工作流. 第一步:使用建模软件导出FBX 1.maya用户 以maya 2020为例,选中想要导出的模型,依次点击文件-游戏导出器,按照如下设置 ...
- tensorflow+入门笔记︱基本张量tensor理解与tensorflow运行结构与相关报错
欢迎登陆官网(附https://tensorflow.google.cn/)了解更多 TensorFlow 内容,也可关注 TensorFlow 官方公众号获取更多资讯. Gokula Krishna ...
最新文章
- ACE-Task结构介绍（二）——消息块ACE_Message_Block结构的分析
- object.defineProperty
- 体验 vue cli 3.0
- 分享codeigniter 路由(URL)终极优化
- matlab 同一坐标系 散点图 t,matlab上机练习
- 网页连接数据库 服务器,关于asp网页连接远程服务器上数据库问题
- 第三章：使用email-ext替换Jenkins的默认邮件通知
- 重装上阵两个人合体机器人_又一部热门定档！林超贤、彭于晏领衔《紧急救援》重装定档2021年大年初一...
- 模式识别之特征提取算法
- 2023南京审计大学计算机考研信息汇总
- cmd命令行激活win7
- 外贸建站自己买主机空间好吗
- “前首富”牟其中：欠的人情太多，要留着命来还
- U盘Windows PE 安装系统简易教程
- 风口下的追逐：AI正在驾驶、客服、教育领域疾驰
- 天大18年c语言离线作业,2018春 Python语言程序设计（天津大学仁爱学院）-中国大学mooc-题库零氪...
- Banner图片轮播器实现ViewPager图片切换效果及下方小圆点
- Realm在android的应用
- 霸王级”寒潮来袭 神华国华“智能供热”送温暖
- [POI2008]Mirror Trap