Jetson Orin Nano安裝torch torchvision

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



# 準備好2個文件放在Dir裏DIR=/home/$USER/Documents/torch2.1

1. torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl 下載地址 https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048 注意JetPack版本，我使用的是5.1.3

2. vision-0.16.1.tar.gz  下載地址  https://github.com/pytorch/vision/releases

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

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev 

#sudo apt-get install libjpeg-dev libpng-dev 

pip3 install 'Cython<3' 

pip3 install numpy 

export DIR=/home/$USER/Documents/torch2.1

cd $DIR
pip3 install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl


tar -zxvf vision-0.16.1.tar.gz

cd vision-0.16.1

python3 setup.py install --user

# 驗證
#cd ../  必須要離開vision的目錄
#python3
#import torchvision
#print(torchvision.__version__)

#  版本對應
#    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 - torchvision v0.15.1
#    PyTorch v2.1 - torchvision v0.16.1
#    PyTorch v2.2 - torchvision v0.17.1
#    PyTorch v2.3 - torchvision v0.18.0