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

一、安装jetpack

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

二、安装jtop

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

重启后,终端(ctrl+alt+t)中输入jtop即可看到各种信息
在这里插入图片描述

三、安装cmake

sudo cmake --version

提示sudo: cmake: command not found
安装:sudo apt-get -y install cmake
不在意cmake版本,能用就行
四、安装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的环境算是初步搭建好了。
在这里插入图片描述

持续更新中。。。

  • 5
    点赞
  • 39
    收藏
    觉得还不错? 一键收藏
  • 18
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值