Jetson Nano & TX2 （五）pytorch

对应版本：

• 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.5.0

• JetPack 4.4 DP
  • Python 3.6 - torch-1.5.0-cp36-cp36m-linux_aarch64.whl
  • As per the PyTorch Release Notes 47, Python 2 is not longer supported

PyTorch v1.4.0

• JetPack 4.4 DP

  • Python 2.7 - torch-1.4.0-cp27-cp27mu-linux_aarch64.whl
  • Python 3.6 - torch-1.4.0-cp36-cp36m-linux_aarch64.whl

• JetPack 4.2 / 4.3

  • Python 2.7 - torch-1.4.0-cp27-cp27mu-linux_aarch64.whl
  • Python 3.6 - torch-1.4.0-cp36-cp36m-linux_aarch64.whl

PyTorch v1.3.0

• JetPack 4.4 DP

  • Python 2.7 - torch-1.3.0-cp27-cp27mu-linux_aarch64.whl
  • Python 3.6 - torch-1.3.0-cp36-cp36m-linux_aarch64.whl

• JetPack 4.2 / 4.3

  • Python 2.7 - torch-1.3.0-cp27-cp27mu-linux_aarch64.whl
  • Python 3.6 - torch-1.3.0-cp36-cp36m-linux_aarch64.whl

PyTorch v1.2.0

• JetPack 4.4 DP

  • Python 2.7 - torch-1.2.0-cp27-cp27mu-linux_aarch64.whl
  • Python 3.6 - torch-1.2.0-cp36-cp36m-linux_aarch64.whl

• JetPack 4.2 / 4.3

  • Python 2.7 - torch-1.2.0a0+8554416-cp27-cp27mu-linux_aarch64.whl
  • Python 3.6 - torch-1.2.0a0+8554416-cp36-cp36m-linux_aarch64.whl

PyTorch v1.1.0

• JetPack 4.2 / 4.3
  • Python 2.7 - torch-1.1.0-cp27-cp27mu-linux_aarch64.whl
  • Python 3.6 - torch-1.1.0-cp36-cp36m-linux_aarch64.whl

PyTorch v1.0.0

• JetPack 4.2 / 4.3
  • Python 2.7 - torch-1.0.0a0+bb15580-cp27-cp27mu-linux_aarch64.whl
  • Python 3.6 - torch-1.0.0a0+bb15580-cp36-cp36m-linux_aarch64.whl

e.g:

PyTorch （Python 3.6）

wget https://nvidia.box.com/shared/static/ncgzus5o23uck9i5oth2n8n06k340l6k.whl -O torch-1.4.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.4.0-cp36-cp36m-linux_aarch64.whl

torchvision

$ sudo apt-get install libjpeg-dev zlib1g-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
$ cd torchvision
$ sudo python setup.py install
$ cd ../  # attempting to load torchvision from build dir will result in import error
$ pip install 'pillow<7' # not needed for torchvision v0.5.0+

 

其他:

针对 aarch 官方 给出 pytorch 安装：地址