栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

tensorRT使用技巧

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

tensorRT使用技巧

一、参考资料 二、使用技巧 1. 如何优化 Python API

How Do I Optimize My Python Performance?

  1. pycuda 支持 Python Buffer Protocol,即允许高效地访问内存数据。

    protobuf全称为ProtoBuffer,是由Goodle开发的一种可以实现内存与硬盘文件交换的协议接口,说白了就是一种可以我们自己定规则协议的一套工具,我们通过自己编写统一的参数描述文件proto,然后利用protoc编译就能生成可以读取信息的代码,非常方便,其中ONNX的协议就是使用Protobuf开发的。
    
2. 测试tensorRT性能指标
  1. Tool:trtexec

    If you have a model saved as an onNX file, or if you have a network description in a Caffe prototxt format, you can use the trtexec tool to test the performance of running inference on your network using TensorRT.
    
  2. CPU Timing

  3. CUDA Events

  4. Built-In TensorRT Profiling

  5. CUDA Profiling

  6. Memory

3. tensorRT模型转换
  1. Caffe的转换器 Caffe-Parse,Tensorflow 的转换器,都有些落后了,不推荐使用
  2. 推荐 ONNX2TensorRT,TensorRT对ONNX的支持最好。
  3. 在深度学习框架中,TensorRT对Pytorch的支持更为友好。
  4. 理论上 95%的模型 都可以转换为TensorRT,条条大路通罗马嘛。只不过有些模型可能转换的难度比较大。如果遇到一个无法转换的模型,先不要绝望,再想想,再想想,看看能不能通过其他方式绕过去。
4. tensorRT plugin

实现TensorRT自定义插件(plugin)自由!

5. 用tensorRT提高模型精度

How Do I Improve My Model Accuracy?

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/283618.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号