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左右