请问用TensorRT扩展实现RetinaNet的后处理部分时如何编译Custom Operator[阿里云机器学习PAI]

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

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
=====这是一个广告位,招租中,联系qq 78315851====
1 条回复 A 作者 M 管理员
  1. 您好,有三种编译Custom Operators的方式:Building with CMake、Building with JIT Compilation及Building with Setuptools。这三种编译方式适用于不同场景,您可以根据自己的需求进行选择。

    1. 首先,需要安装TensorRT的开发环境,包括CUDA、CUDNN、TensorRT等。

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

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