Jetson配置yolov5运行环境

发表于

Jetson yolov5

ARM aarch64 architecture

PyTorch v1.10.0

  • sudo apt-get update
  • sudo apt-get install python3-dev
1
2
3
4
5
wget https://nvidia.box.com/shared/static/fjtbno0vpo676a25cgvuqc1wty0fkkg6.whl -O torch-1.10.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy   
pip3 install torch-1.10.0-cp36-cp36m-linux_aarch64.whl

(根据机器性能可能要等待比较久)

  • torchvision
1
2
3
4
5
6
7
8
9
10
11
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

git clone --branch v0.11.1 https://github.com/pytorch/vision torchvision


cd torchvision
export BUILD_VERSION=0.11.1

python3 setup.py install --user

cd ../