1. 下载yolov5

git clone https://github.com/ultralytics/yolov5.git
cd yolov5-master

2. 安装依赖

pip3 install -r requirements.txt

3. 卸载torch & torchvision

pip3 uninstall torch torchvision

4. 安装jetson xavier 相应版本的torch torchvision

官方下载

5. 验证cuda

import torch
torch.cuda.is_available()
#True

6. 运行yolov5

python3 detect.py --source data/images/bus.jpg --weights yolov5s.pt

可以看到在使用yolov5s且使用cuda的情况下640x640分辨率的图片识别速度在50ms左右