Jetson Nano YOLOv5 Pytorch 설치하기!

(이 문서는 qengineering.eu를 참고하였습니다.)

우선 Jetson Nano에 깔려 있는
Python 버전은 3.6.9입니다.

Terminal에서 파이썬 버전 확인 : python3 -V

이에 맞는 PyTorch를 먼저 고르고 설치해야
Torchvision과 YOLOv5를 설치하실 수 있습니다.

출처 : PyPi.org

Jetson Nano에 맞게
PyTorch 1.10.0a0+ggit36449ea
Torchvision 0.11.0a0_fa347eb
로 설치하겠습니다.

---

설치 방법 1. wheel (추천!)

(간편한 복붙을 위해 $는 생략하겠습니다!)

• 의존성 문제 방지를 위한 설치

sudo apt-get install python3-pip libjpeg-dev libopenblas-dev libopenmpi-dev libomp-dev
sudo -H pip3 install future
sudo pip3 install -U --user wheel mock pillow
sudo -H pip3 install testresources
sudo -H pip3 install setuptools==58.3.0
sudo -H pip3 install Cython

(setuptools는 해당 버전보다 높다면 의존성 문제가 발생!)

• Google drive에 게시된 PyTorch 1.10.0 설치

sudo -H pip3 install gdown
gdown https://drive.google.com/uc?id=1TqC6_2cwqiYacjoLhLgrZoap6-sVL2sd
sudo -H pip3 install torch-1.10.0a0+git36449ea-cp36-cp36m-linux_aarch64.whl

• PyTorch 삭제

rm torch-1.10.0a0+git36449ea-cp36-cp36m-linux_aarch64.whl

---

설치 방법 2. Git Clone

(간편한 복붙을 위해 $는 생략하겠습니다!)

• 의존성 문제 방지를 위한 설치

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install ninja-build git cmake
sudo apt-get install libjpeg-dev libopenmpi-dev libomp-dev ccache
sudo apt-get install libopenblas-dev libblas-dev libeigen3-dev
sudo pip3 install -U --user wheel mock pillow
sudo -H pip3 install testresources
sudo -H pip3 install setuptools==58.3.0
sudo -H pip3 install scikit-build

• Git을 통한 PyTorch 설치

git clone -b v1.10.0 --depth=1 --recursive https://github.com/pytorch/pytorch.git
cd pytorch
sudo pip3 install -r requirements.txt

(이 후 공간 확보를 위해 PyTorch 폴더를 설치해주면 됩니다.)

---

설치 확인

Terminal 접속
> Python3 실행 python3
> import torch
> torch.__version__
> 버전 확인 후 exit() 으로 python3 종료

---

마치며

Git Clone 방식은 requirements.txt를 설치할 때,
여러 의존성 문제
(이미 YOLOv5 현재 버전은 Jetson Nano의 Python 3.6.9를 지원하지 않아)
충돌되고 오류가 나는 경우가 많습니다.

wheel 설치를 이용하자!
만약 Git Clone을 이용한다면 의존성을 다 확인하자!

---

이어서

Torchvision 설치하기