Jetson TX2安装pytorch(from binaries)

下载

• Stable binaries:
  • Python 2.7: https://nvidia.box.com/v/torch-stable-cp27-jetson-jp42
  • Python 3.6: https://nvidia.box.com/v/torch-stable-cp36-jetson-jp42
• Rolling weekly binaries:
  • Python 2.7: https://nvidia.box.com/v/torch-weekly-cp27-jetson-jp42
  • Python 3.6: https://nvidia.box.com/v/torch-weekly-cp36-jetson-jp42

安装

进如whl文件保存的目录，以stable binaries中python3.6版本为例，输入以下指令：

sudo pip3 install torch-1.0.0a0+bb15580-cp36-cp36m-linux_aarch64.whl

如果报错：zipfile.BadZipFile: File is not a zip file
大概率是whl文件下载不完整，重新下载就能解决

安装torchvision

sudo pip3 install torchvision

参考文献

https://github.com/pytorch/pytorch