一、安装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的环境算是初步搭建好了。
持续更新中。。。