上传者: jacke121
|
上传时间: 2022-07-03 21:08:33
|
文件大小: 221.22MB
|
文件类型: RAR
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