一.首先是捋请思路
①刷机后的Orin上环境是:CUDA11.4+CUDNN8.3.2使得后续需要的部署环境只能为TensorRT8.x(这是根据cuda和cudnn的版本确定的,安装见:三.TensorRT加速优化(1))
②TensorRT部署这里需要一个版本对应可以看到tensorrtx的最高yolov5支持的是v5.0模型,所以第③步最高选择是到YOLOv5_v5.0.下载:(GitHub - wang-xinyu/tensorrtx at yolov5-v5.0)

③ YOLOv5_v5.0下载:(https://github.com/ultralytics/yolov5/tree/v5.0)
④ 下载.pt文件,有yolov5s.pt / yolov5l.pt / yolov5m.pt ....等,这里下载yolov5s.pt(对应上面的YOLOv5_v5.0)下载:https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt
⑤ 需要有OpenCV,我安装的是opencv-4.6.0 : 安装OpenCV4.6.0简洁过程--Jetson AGX Orin
二.参考Github上的步骤流程
1. generate .wts from pytorch with .pt, or download .wts from model zoo
git clone -b v5.0 https://github.com/ultralytics/yolov5.git
git clone https://github.c

本文详细介绍了如何在Orin平台上使用CUDA11.4和CUDNN8.3.2环境通过TensorRT部署YOLOv5_v5.0模型,从刷机环境配置、模型下载、权重转换到实际部署过程,展示了TensorRT在图像检测中的性能优化效果。
1139

被折叠的 条评论
为什么被折叠?



