当前位置: 首页 > news >正文

TensorRT的结构

Builder(网络原数据):模型搭建的入口,网络的tensorRT内部表示以及可执行程序引擎都是由该对象的成员方法生成的

        BuiderConfig(网络原数据的选项):负责设置模型的一些参数,如是否开始fp16模式,in8模式等等

        Network(计算图内容):网络的主体,在使用api搭建网络的过程当中,将不断地向其中添加一些层,并标记网络的输入输出节点,在其他两种workflow中,则使用Parser来加载来自Onnx文件中的网络,就无需一层一层手工添加

        SerializedNetwork:模型网络在TRT中的内部表示,可用它生成可执行的推理引擎或者把它序列化保存为文件,方便以后读取和使用

        Engine:推理引擎,是模型计算的核心,可理解为可执行程序的代码段

        context:用于计算GPU的上下文。类比于cpu上的进程概念进程,是执行推理引擎的主体

        buffer相关(数据的内存、显存相关):我们将需要把数据从cpu端搬到gpu端,执行推理计算后,再把数据由gpu端搬回cpu端。这些步骤涉及内存的申请、拷贝和释放,需要借助CUDA库或者扩大python库才能完成

        excute:调用计算核心执行推理计算的具体过程 

        善后工作:还有一些外围的工作需要手工去完成

http://www.lryc.cn/news/188116.html

相关文章:

  • python对excel数据表进行数据清洗
  • 95、Spring Data Redis 之使用RedisTemplate 实现自定义查询 及 Spring Data Redis 的样本查询
  • jdbc(DriverManager+Connection+Statement+ResultSet)+SQL注入+开启预编译+数据连接池
  • NoSQL之 Redis命令工具及常用命令
  • 多线程(线程互斥)
  • 使用 html2canvas 和 jspdf 将页面转 pdf,同时解决当页面过长时,页面白屏问题
  • 【Python 千题 —— 基础篇】今年几岁啦
  • git push 失败 shallow update not allowed
  • uniapp 在uni.scss 根据@mixin定义方法 、通过@include全局使用
  • C++ 类和对象(一)
  • rust函数
  • 链表的基本操作
  • Flutter AI五子棋
  • springboot项目中后台文件上传处理
  • 【SQL】MySQL中的存储引擎、事务、锁、日志
  • DRM全解析 —— CRTC详解(2)
  • 3d环形图开发(vue3+vite+ts)
  • element ui中父子组件共用一个el-dialog弹窗,切换组件页面弹窗进行关闭
  • 基于Keil a51汇编 —— Segments, Modules, and Programs
  • 基于Java+SpringBoot+Vue民宿管理系统的设计与实现 前后端分离【Java毕业设计·文档报告·代码讲解·安装调试】
  • 【Qt】三种方式实现抽奖小游戏
  • 【算法与数据结构】--算法基础--算法设计与分析
  • vue部分入门知识点代码示例
  • 【图灵】Spring为什么要用三级缓存解决循环依赖问题
  • bert入门
  • 蓝桥杯基础---切面条
  • Spring Data Redis使用方式
  • HarmonyOS/OpenHarmony原生应用开发-华为Serverless认证服务说明(二)
  • 华为数通方向HCIP-DataCom H12-831题库(多选题:241-259)
  • 深度学习在 NumPy、TensorFlow 和 PyTorch 中实现所有损失函数