使用c++部署tensorrt加速yolov7

        先放上一张我运行成功的截图,只要跟着我的教程一步一步按操作,下载好匹配的软件是一定可以成功的!使用c++部署tensorrt加速yolov7_第1张图片

       我相信想要在C++平台使用tensorrt加速的朋友们也是有很强的计算机基础的,那么简单的部分我们就跳过,重点是和大家介绍模型转换的部分以及环境的搭建。

一. 环境

我的cudn版本

使用c++部署tensorrt加速yolov7_第2张图片

我的cudnn版本是8.1.1 那么我下载的对应的tensorrt版本是7.2.2.3 ,当然并不是要严格安装我的版本来,但是如果和我的版本一样那么成功的概率会很大。具体tensorrt支持的cuda和cudnn可以在官网上查看。里面有支持的各种版本,网上安装tensorrt的教程很多这里就不多介绍了。

tensorrt下载官网

二. yolov7模型的导出

       我想要重点强调的就是这里的onnx模型不能使用官网的代码直接导出因为输出维度不对,需要进行修改!!!

你可能感兴趣的:(深度学习,人工智能)