tongchenkeji 发表于:2023-1-29 13:11:260次点击 已关注取消关注 关注 私信 请问用TensorRT扩展实现RetinaNet的后处理部分时如何编译Custom Operator[阿里云机器学习PAI] 暂停朗读为您朗读 请问用TensorRT扩展实现RetinaNet的后处理部分时如何编译Custom Operators? 「点点赞赏,手留余香」 赞赏 还没有人赞赏,快来当第一个赞赏的人吧! 海报 机器学习PAI# 人工智能平台 PAI1410
晨光永不消逝AM 2023-11-28 2:50:22 1 您好,有三种编译Custom Operators的方式:Building with CMake、Building with JIT Compilation及Building with Setuptools。这三种编译方式适用于不同场景,您可以根据自己的需求进行选择。
认真学习的heartAM 2023-11-28 2:50:22 2 首先,需要安装TensorRT的开发环境,包括CUDA、CUDNN、TensorRT等。 然后,需要编写RetinaNet的后处理部分的Custom Operator,并将其编译成可以在TensorRT中使用的库文件。 最后,需要将Custom Operator库文件加载到TensorRT中,并将其与RetinaNet的后处理部分进行集成。
您好,有三种编译Custom Operators的方式:Building with CMake、Building with JIT Compilation及Building with Setuptools。这三种编译方式适用于不同场景,您可以根据自己的需求进行选择。
首先,需要安装TensorRT的开发环境,包括CUDA、CUDNN、TensorRT等。
然后,需要编写RetinaNet的后处理部分的Custom Operator,并将其编译成可以在TensorRT中使用的库文件。
最后,需要将Custom Operator库文件加载到TensorRT中,并将其与RetinaNet的后处理部分进行集成。