젯슨 나노 오린에 파이토치 설치하기

케이스 안에 숨었다.

나노 오린에 jetpack5.1을 설치하였다.

OS는 우분투 20.04가 설치되었고, 파이토치를 설치하려 했으나 실패하였다.

검색해 보니 오린에 맞는 파이토치는 nvidia에서 별도로 제공하였다.

https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

PyTorch for Jetson

jetpack 버전에 맞는 것을 설치하면 된다.

설치는 해당 파일을 다운로드 받아서 스크립트를 통해 직접 설치했다.

$ export TORCH_INSTALL=/home/exotech/Desktop/torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
$ pip install --no-chache $TORCH_INSTALL torchvision

하지만 torchvision은 정상적으로 동작하지 않았다. 0.14버전이 토치 2.1과는 호환되지 않는 오류였다.

WARNING ⚠️ torchvision==0.14 is incompatible with torch==2.1.
Run 'pip install torchvision==0.16' to fix torchvision or 'pip install -U torch torchvision' to update both.
For a full compatibility table see https://github.com/pytorch/vision#installation

그래서 비전을 0.16으로 강제설치했다.

pip install torchvision=0.16

그랬더니 비전은 정상적으로 0.16이 설치되었으나, 토치가 CPU버전의 2.1로 재설치되어 버렸다.

 

이야기가 좀 길어졌는데, 결론부터 얘기하자면 위 토치를 설치할 때 비전은 설치하지 않는다.

그리고 비전은 그냥 설치하면 0.14로 설치하기 때문에 0.16버전을 직접 다운로드 받아 빌드하여 설치한다.

 

다시 토치만 설치해보자.

$ export TORCH_INSTALL=/home/exotech/Desktop/torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl
$ pip install --no-chache $TORCH_INSTALL

 

토치비전을 다운로드 받는다.

$ git clone --branch v0.16.0 https://github.com/pytorch/vision.git

소스코드 안에서 아래를 실행한다.

$ pip install -U setuptools
$ python setup.py install

비전설치는 아래를 참조

https://makejarvis.tistory.com/166

torch / vision 설치 on Jetson Xavier (yolo GPU학습용)

버전정보는 아래에서 확인할 수 있다.

import torch
print("PyTorch version:", torch.__version__)
print("CUDA available:", torch.cuda.is_available())
print("CUDA version:", torch.version.cuda)

 

이제 가상환경에서 torch 및 torchvision을 설치해 보자.

가상환경은 미니콘다를 이용했다.

jetpack을 이용해 설치하면 파이썬은 3.8 버전이 설치된다.

가상환경을 jetpack에 파이썬 버전 그대로 설치하게 되면 가상환경에서 별도의 설치 없이 바로 torch를 사용할 수 있다.

$ conda create -n asda python=3.8
$ conda activate asda
$ pip list

(asda) exotech@ubuntu:~/webapps$ pip list
Package               Version
--------------------- -----------------------
apt-clone             0.2.1
apturl                0.5.2
asgiref               3.8.1
.
.
중략
.
.
tensorrt              8.5.2.2
tomli                 2.0.1
torch                 2.1.0a0+41361538.nv23.6
torchvision           0.16.0+fbb4cc5
tqdm                  4.67.1
typeguard             4.3.0
typing_extensions     4.12.2
tzdata                2024.2
ubuntu-drivers-common 0.0.0
ubuntu-pro-client     8001
uff                   0.6.9
ultralytics           8.3.58
ultralytics-thop      2.0.13
urllib3               2.2.3
urwid                 2.0.1
wadllib               1.3.3
wheel                 0.43.0
xkit                  0.0.0
zipp                  3.20.2

하지만 jetpack과  다른 버전의 파이썬을 설치하면 torch를 사용할 수 없다.

$ conda create -n test python=3.7
$ conda activate test
$ pip list

(test) exotech@ubuntu:~/webapps$ pip list
Package    Version
---------- -------
pip        24.0
setuptools 69.0.3
wheel      0.42.0

그럼 가상환경에서 토치를 설치해 보자.

(test) exotech@ubuntu:~/webapps$ pip install --no-cache $TORCH_INSTALL
ERROR: torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl is not a supported wheel on this platform.

가상환경에서의 설치는 플랫폼 자체에서 지원하지 않는 것 같다.