开发者社区 > 大数据与机器学习 > 人工智能平台PAI > 正文

请问用TensorRT扩展实现RetinaNet的后处理部分时如何编译Custom Operator

已解决

请问用TensorRT扩展实现RetinaNet的后处理部分时如何编译Custom Operators?

展开
收起
felix@ 2023-01-29 13:11:26 2015 3
2 条回答
写回答
取消 提交回答
  • 今天也要加油吖~
    采纳回答

    您好,有三种编译Custom Operators的方式:Building with CMake、Building with JIT Compilation及Building with Setuptools。这三种编译方式适用于不同场景,您可以根据自己的需求进行选择。

    2023-01-29 13:13:16
    赞同 1 展开评论 打赏
    1. 首先,需要安装TensorRT的开发环境,包括CUDA、CUDNN、TensorRT等。

    2. 然后,需要编写RetinaNet的后处理部分的Custom Operator,并将其编译成可以在TensorRT中使用的库文件。

    3. 最后,需要将Custom Operator库文件加载到TensorRT中,并将其与RetinaNet的后处理部分进行集成。

    2023-01-29 15:50:28
    赞同 2 展开评论 打赏

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载