Jetson Nano YOLOv5 Torchvision 설치하기

해당 사진을 보면 이미 설치한 PyTorch와 맞는 버전의
Torchvision을 찾을 수 있습니다!

저희는 이전에 PyTorch 1.10.0을 설치하였다 가정하고,
Torchvision 0.11.0을 설치하겠습니다.

설치하기

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

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

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev
sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev
sudo pip3 install -U pillow

• Google Drive를 통해 Torchvision 0.11.0 설치

sudo -H pip3 install gdown
gdown https://drive.google.com/uc?id=1C7y6VSIBkmL2RQnVy8xF9cAnrrpJiJ-K
sudo -H pip3 install torchvision-0.11.0a0+fa347eb-cp36-cp36m-linux_aarch64.whl

• Torchvision 파일 삭제를 통한 공간 확보

rm torchvision-0.11.0a0+fa347eb-cp36-cp36m-linux_aarch64.whl

설치 확인

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

마치며

OpenCV, PyTorch, Torchvision 모두 설치 완료했다면,
YOLOv5를 Clone하여 동작시켜보도록 하겠습니다!

Torchvision과 PyTorch 버전을 꼭 확인하고 설치하는 버릇!