【tensorrt】——插件写法及python plugin大体流程
发布于2021-07-25 07:39 阅读(1285) 评论(0) 点赞(18) 收藏(0)
tensorrt
1. python plugin 大体流程
这是参考torch2trt写的一个python插件的大体流程，因为tensorrt是没有开源的，所以具有不确定性。
2. code
项目，是做tensorrt测试的，包括c++/python api。因为tensorrt模型的构建不是api就是转换模型之后直接推理。如果想单独测试某个layer就需要这样的小项目。
reference
原文链接:https://blog.csdn.net/u011622208/article/details/119041630
所属网站分类: 技术文章 > 博客
作者：空气很好
链接：http://www.pythonpdf.com/blog/article/480/4e8b9d2e9885fdb65c17/
来源：编程知识网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容：(最多支持255个字符)