tensorrt c++ 推理示例项目,支持分类网络,比如alexnet,mobileone,skipnet等轻量级网络。
环境:win10 vs2017环境,cuda11.0,亲测成功。
tensorrt库版本:
TensorRT-8.2.1.8.Windows10.x86_64.cuda-11.4.cudnn8.2
转换流程:pytorch转onnx,onnx转tensorrt引擎
python生成tensorrt引擎步骤,可以查看我的博客:
https://blog.csdn.net/jacke121/article/details/125382721
1