利用TensorRT在jetson orin 上加速YOLOv5

一、第一种方法，需要下载各种包：

要用到一个大佬的开源，GitHub地址如下：

https://github.com/wang-xinyu/tensorrtx/tree/master/yolov5

1. 安装pycuda，在线安装pycuda

pip3 install pycuda

或者：

sudo pip3 install --global-option=build_ext --global-option="-I/usr/local/cuda-10.2/targets/aarch64-linux/include/" --global-option="-L/usr/local/cuda-10.2/targets/aarch64-linux/lib/" pycuda==2019.1

2. Windows操作（为了得到加速后的wts)

1.YOLOv5-7.0的原版的开源程序
2. 将大佬开源的项目tensorrtx，下载到自己的windows电脑上
然后，把tensorrtx文件夹整体，复制粘贴到yolov5-7.0原版程序的文件夹中。再把tensorrtx-yolov5-v7.0\yolov5\gen_wts.py复制粘贴到yolov5原版文件夹中。

然后打开终端，激活在anaconda中自己创建的虚拟环境，输入命令：

python gen_wts.py -w yolov5n-seg.pt -o yolov5n-seg.wts

因为我是需要Y