nvidia的官网安装指导说明PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums

Select the version of torchvision to download depending on the version of PyTorch that you have installed:

 PyTorch v1.12.0

 PyTorch v1.11.0

 PyTorch v1.10.0

 PyTorch v1.9.0

 PyTorch v1.8.0

 PyTorch v1.7.0

 PyTorch v1.6.0

 PyTorch v1.5.0

 PyTorch v1.4.0

 PyTorch v1.3.0

 PyTorch v1.2.0

 PyTorch v1.1.0

 PyTorch v1.0.0

指示

安装

Below are example commands for installing these PyTorch wheels on Jetson. Substitute the URL and filenames from the desired PyTorch download from above.

> Python 3.6

wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.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.8.0-cp36-cp36m-linux_aarch64.whl

> Python 2.7

wget https://nvidia.box.com/shared/static/1v2cc4ro6zvsbu0p8h6qcuaqco1qcsif.whl -O torch-1.4.0-cp27-cp27mu-linux_aarch64.whl
sudo apt-get install libopenblas-base libopenmpi-dev 
pip install future torch-1.4.0-cp27-cp27mu-linux_aarch64.whl

(PyTorch v1.4.0 for L4T R32.4.2 is the last version to support Python 2.7)

> torchvision

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.x.0  # where 0.x.0 is the torchvision version  
$ python3 setup.py install --user
$ cd ../  # attempting to load torchvision from build dir will result in import error
$ pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

Select the version of torchvision to download depending on the version of PyTorch that you have installed:

• 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

 

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.x.0  # where 0.x.0 is the torchvision version  
$ python3 setup.py install --user