https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048
PyTorch for Jetson
Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson
forums.developer.nvidia.com
해당 페이지에서 본인의 jetpack 버전에 맞는 pytorch 배포본 다운로드 할것
>> wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.10.0-cp36-cp36m-linux_aarch64.whl
>> sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
>> pip3 install Cython
>> pip3 install numpy torch-1.10.0-cp36-cp36m-linux_aarch64.whl
설치 후 적합한 torchvision 버전을 위 페이지에서 다시 찾을 것
- PyTorch v1.0 - torchvision v0.2.2
- PyTorch v1.1 - torchvision v0.3.0
- PyTorch v1.2 - torchvision v0.4.0
- PyTorch v1.3 - torchvision v0.4.2
- PyTorch v1.4 - torchvision v0.5.0
- PyTorch v1.5 - torchvision v0.6.0
- PyTorch v1.6 - torchvision v0.7.0
- PyTorch v1.7 - torchvision v0.8.1
- PyTorch v1.8 - torchvision v0.9.0
- PyTorch v1.9 - torchvision v0.10.0
- PyTorch v1.10 - torchvision v0.11.1
- PyTorch v1.11 - torchvision v0.12.0
- PyTorch v1.12 - torchvision v0.13.0
- PyTorch v1.13 - torchvision v0.13.0
- PyTorch v1.14 - torchvision v0.14.1
- PyTorch v2.0.0 - torchvision v0.15.1
torchvision 설치 (python 3.6)
$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch v.0.xx.x https://github.com/pytorch/vision torchvision
$ cd torchvision
$ export BUILD_VERSION=0.xx.x
$ python3 setup.py install --user
$ cd ..