Jetson Orin nx 8gb에 pytorch 설치하기

Jetpack Version : 5.1.1

Ubuntu 20.04

arm64 아키텍처

 

최근에 NVIDIA Jetson Orin™ NX 8GB를 구매 했고, pytorch가 설치되어 있지 않아서 설치하는 과정을 공유하고자 한다.

https://www.icbanq.com/P015430888

[110110185]reComputer J4011-Edge AI Device with NVIDIA Jetson Orin™ NX 8GB module, 4xUSB 3.2, M.2 Key E & Key M Slot, Aluminum

 

패키지 설치

$ package=torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl
$ wget "https://developer.download.nvidia.com/compute/redist/jp/v511/pytorch/${package}"

 

conda를 활용하여  python3.8 환경을 만들어 준 다음에 설치를 진행한다.

참고) python3.8 가상환경 만드는 명령어

conda create -n <name> python=3.8

 

 

관련 패키지 설치

$ pip3 install idna certifi decorator pexpect urllib3

 

 

pytorch 설치

위에서 설치한 파이토치 패키지를 pip을 이용하여 설치해준다

$ pip3 install --no-cache "${package}"

 

 

 

결과 확인

$ python -c '
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.empty((1, 2), device=torch.device("cuda")))
'

 

정상적으로 설치된 것을 알 수 있다.

 

 

 

<Reference>

https://pytorch.org/audio/stable/build.jetson.html

https://docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html