在Jetson Xavier NX配置Yolov5s环境

最新推荐文章于 2023-06-09 13:05:27 发布

jhunmk29

最新推荐文章于 2023-06-09 13:05:27 发布

阅读量193

点赞数

文章标签： YOLO python linux

本文链接：https://blog.csdn.net/qq_36070656/article/details/129679994

版权

准备工作

查看系统
在这里插入图片描述
查看内核版本

查看CPU

系统存储

sudo parted -l

在这里插入图片描述
存储空间

df -h

在这里插入图片描述
运行中进程查看

top

python版本查看
在这里插入图片描述
CUDA Version

实时查看板子

sudo -H pip3 install jetson-stats
sudo jtop

pytorch安装

Pytorch-nvidia
查看jetpack版本

sudo jtop

在这里插入图片描述
选择对应版本

wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

torchvision安装

注意版本匹配：https://github.com/pytorch/vision/tree/v0.14.1

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.x.0  # where 0.x.0 is the torchvision version  
$ python3 setup.py install --user
$ cd ../  # attempting to load torchvision from build dir will result in import error
$ pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

可能会报错：

SyntaxError: future feature annotations is not defined

在安装torchvision的时候会自动安装 pillow 9.2.0版本，这时候pillow 的版本过高，卸载pillow重新装8.4.0版本

pip3 install pillow==8.4.0

测试

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

opencv-pyhon安装

清除opencv

edit@edit-desktop:~/YOLOV5$ sudo apt-get purge libopencv*
edit@edit-desktop:~/YOLOV5$ sudo apt autoremove
edit@edit-desktop:~/YOLOV5$ sudo apt-get update

相关依赖

edit@edit-desktop:~/YOLOV5$ sudo apt-get install build-essential
edit@edit-desktop:~/YOLOV5$ sudo apt-get install libglew-dev libtiff5-dev zlib1g-dev libjpeg-dev \ 
	libavcodec-dev libavformat-dev libavutil-dev libpostproc-dev \
	libswscale-dev libeigen3-dev libtbb-dev libgtk2.0-dev pkg-config libpng-dev	
edit@edit-desktop:~/YOLOV5$ sudo apt-get install libatlas-base-dev gfortran
edit@edit-desktop:~/YOLOV5$ sudo apt update
edit@edit-desktop:~/YOLOV5$ sudo apt upgrade