pytorch 自定义算子 onnx1.9 tensorrt8.2

Posted 2023-02-21

参考技术A TensorRT repo

参考 csdn

在 TensorRT/samples/python/ 下参考 costom 编写方式和cmake，docker环境里make生成 .so, 注意插件名字和trt用到的组件名应该一致

python 加载插件

如果用 trtexec ，需要 vim .bashrc

安装依赖 参考