search

dlunion / tensorRTIntegrate

TensorRT ONNX Plugin、Inference、Compile
464 stars 114 forks source link

请问一下，注册DCNv2插件，在完成dladcn_export_onnx.py后该执行哪个步骤呢？ #44

Open shengjie-shi opened 3 years ago

shengjie-shi commented 3 years ago

你好，我想请问一下，我想将.pth模型转成tenorRT的.engine文件，根据提示完成了dladcn_export_onnx.py中导出.onnx的步骤，但是对于注册DCNv2操作还是不太明白。我应该再执行src/onnxplugin/plugins/DCNv2.cu文件吗？即执行： nvcc -c src/onnxplugin/plugins/DCNv2.cu 命令吗？