https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048
해당 페이지에서 본인의 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 ..