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,可以看到
jetson orin nx上搭建yolov5的运行环境_第1张图片

三、安装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__)

查看是否安装成功。
查看版本对应关系
jetson orin nx上搭建yolov5的运行环境_第2张图片

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
jetson orin nx上搭建yolov5的运行环境_第3张图片
四、配置yolov5的requirement
从github上下载yolov5,我是用的yolov5-7版本的。
打开requirement,将第9、16、17行用#注释掉。jetson orin nx上搭建yolov5的运行环境_第4张图片

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的环境算是初步搭建好了。
jetson orin nx上搭建yolov5的运行环境_第5张图片

持续更新中。。。

你可能感兴趣的:(yolov5,jetson,orin,nx,YOLO)