Jetson Xavier AGX - Jetpack5.0.1세팅(3) - PyTorch v1.12.0 , torchvision 0.13.0 설치하기

 

◎ PyTorch설치

Pytorch설치같은경우엔

아래 포럼을 참조하여 작성하였습니다.

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-11-now-available/72048

PyTorch for Jetson - version 1.11 now available

위 포럼에 들어가서 보면 Jetpack버전별로 받을 수 있는 PyTorch Wheel파일 다운받을 수 있습니다.

 

 

현재 Jetpack 5.0.1버전을 사용하고 있으므로,

가장 최신버전의 PyTorch 인 v1.12.0을 사용할 수 있습니다.

 

클릭하여 다운받아줍니다.

 

 

 

$ sudo apt update

$ sudo apt-get upgrade

 

 

먼저, 설치전에 의존성 패키지를 다운받아줍니다

 

$ sudo apt-get install python3-pip libopenblas-base libopenmpi-dev

$ pip3 install Cython numpy

 

 

이제 PyTorch를 설치합니다.

$ cd Downloads

$ sudo pip3 install <Torch wheel File>

  * sudo 빼먹으면 안됩니다.

 

 

 

 

◎ Torchvision 설치

이제 Torchvision을 설치해야합니다.

이에 대한 의존성 패키지는

 

$ sudo apt-get install libjpeg-dev zlib1g-dev

 

 

 

이후 아래 github링크로 들어가보면

https://github.com/pytorch/vision

GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision

PyTorch버전에 따른 torchvision 버전이 나와있습니다.

저는 PyTorch 1.12.0버전이므로, TorchVision 0.13.0버전을 깔아보도록 하겠습니다.

 

 

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

$ cd torchvision
$ sudo python3 setup.py install

*여기서 시간이 좀 오래걸립니다

 

 

 

 

$ python3

>>import torch

>>import torchvision

 

아무이상없이 import 된다면 성공