准备虚拟机
楼主虚拟机版本,测试过ubuntu18.04,20.04,(主板长按中间键,同时短按恢复键进入刷机模式,虚拟机检测到usb设备,选择虚拟机链接。)进入刷机模式后要保证USB端口能检测到,如图如果连接状态是灰色,就是检测不到,不用折腾了,换镜像版本,18.04也有多个版本。虚拟机安装正常安装, 硬盘50g左右。
下载SDKManager
官方下载地址:https://developer.nvidia.com/sdk-manager;
虚拟机打开命令终端安装sdkmanager
sudo apt install ./sdkmanager_xxx_amd64.deb
安装后打开
sdkmanager
首次使用需要注册NVIDIA 开发者账号登录后才能使用
sdkmanager Target Hardware会自动检测适配Jetson硬件,以及SDK版本,点击CONTINUE
sdk一起安装容易失败,试过很多次一次没有成功,换版本也不行,最后分开安装,只勾选系统Jetson Linux,勾选accpet接收,点击继续CONTINUE
开始刷机,期间会弹出检查安装硬盘,主板装了固态硬盘ssd卡,选择nvme,等待20多分钟,烧写完成即可。
板端安裝nvidia-jetpack SDK版本
需要有网环境,主板打开终端输入命令安装
sudo apt-get update
sudo apt-get install nvidia-jetpack -y
等待10多分钟安装完成
jtop安装 查看系统版本状态
$ sudo apt install python3-pip
$ sudo pip3 install jetson-stats
$ sudo jtop
sudo apt install libopenblas-base libopenmpi-dev libomp-dev -y
安装yolo环境ultralytics
pip install ultralytics
(默认安装基于CPU的pytorch和vision,需要手动安装对应版本pytorch和vision,这里使用的torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl)
安装pytorch
下载链接
https://developer.download.nvidia.cn/compute/redist/jp/v511/pytorch/torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl
执行pip install torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl
https://github.com/pytorch/vision/tree/release/0.15
cd vision
python setup.py install
转换onnx模型
/usr/src/tensorrt/bin/trtexec --onnx=yolo26s.onnx --saveEngine=yolo26s_fp16.engine --fp16 --workspace=4096 2>&1 | tee trtexec_fp16.log