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