注意:不同用户安装的python可能会在不同的路径,因此不同的pip管理会导致安装的 torch和torchvision会在不同的路径下 记得区分用户来运行yolo
一、确认系统 JetPack 版本 此处使用5.1.1
1、查看JetPack 版本
jtop
二、安装 ultralytics、pytorch、torchvision、onnxruntime-gpu、numpy
1、查看当前已安装的软件
pip list
2、安装 ultralytics 主分支最新版 可以跑旧版本 … v5 v8 v11
1、更新软件包列表,安装 pip 并升级到最新版本
sudo apt update
sudo apt install python3-pip -y
pip install -U pip
2、安装 ultralytics pip 软件包与可选依赖项
pip install ultralytics[export]
3、重启设备
sudo reboot
3、安装 PyTorch 和 Torchvision 版本要求对应 JetPack 来选择
1、卸载当前安装的PyTorch 和 Torchvis