jetson orin nx上搭建yolov5的运行环境

一、安装jtop

sudo apt install python3-pip
sudo -H pip3 install -U jetson-stats
reboot

重启后，终端（ctrl+alt+t）中输入jtop即可看到各种信息

二、安装jetpack

sudo apt-get update
sudo apt-get dist-upgrade
sudo apt-get install nvidia-jetpack

安装完成之后，打开jtop，可以看到

三、安装torch和torchvision

安装torch：
在PyTorch for Jetson选择自己需要的版本的pytorch进行下载，我是安装的v1.12.0版本的。

sudo apt-get install -y libopenblas-base libopenmpi-dev
pip3 install torch-1.12.0a0+2c916ef.nv22.3-cp38-cp38-linux_aarch64.whl

在桌面另起一个终端（ctrl+alt+t），输入

python3
import torch
print(torch.__version__)

查看是否安装成功。
查看版本对应关系

torch版本是v1.12.0
对应torchvision的版本是v0.13.0

安装torchvision：

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.13.0 https://github.com/pytorch/vision torchvision
cd torchvision
export BUILD_VERSION=0.13.0
python3 setup.py install --user

安装需要一段时间。
完成之后，在桌面另起一个终端（ctrl+alt+t），输入：

python3
import torchvision
print(torchvision.__version__)

会显示torchvision的版本，在此终端继续输入如下指令：

python3
import torch
print(torch.cuda.is_available())

返回true

四、配置yolov5的requirement
从github上下载yolov5，我是用的yolov5-7版本的。
打开requirement，将第9、16、17行用#注释掉。

python3 -m pip install --upgrade pip
cd yolov5
pip3 install -r requirements.txt

包安装完之后，

cd yolov5
python detect.py --source data/images --weight yolov5m.pt --view-img

终端中会显示，运行结果保存在 runs/detect/exp中。
yolov5的环境算是初步搭建好了。

持续更新中。。。