前言
昨天下午因为工作拿到了jetson nano的板子，习惯性的尝试安装pytorch，却发现不对劲，经过一番折磨之后终于在今天上午把它安装好了，在这里分享一下经验，希望后来人少走弯路。
话先说在前头：如果你的nano是python3.6.9（对应jetpack 4.6），千万不要尝试升级python版本。nano最高只支持jetpack4.6。
如何安装
环境：jetson nano ubuntu18.04，已经刷好了jetpack4.6，python3是3.6.9
想安装cpu版的torch很简单，直接输pytorch上的官网命令就行。
但如果想用cuda，也就是gpu版本的torch，只有两种方法：
1.自己编译pytorch的源码。
2.下载英伟达官方或者别人预编译好的pytorch的轮子。
这边我们只说第二种方法：
链接1包含torch和torchvision
链接2只含torch
另外由于文件过大，都是放在谷歌云盘里的，怎么上谷歌云盘就不多说了。
下载的时候注意，jetson nano只能下3.6python支持的，也就是最高支持pytorch的1.10.0版本，对应torchvision的版本是0.11.0。为什么呢？是因为
Hi @m.ahmad4984, JetPack 5.0.2 is in the final preparations to be released, so it should be posted in the next few days pending any unforeseen delays or blockers.
As per the Jetson Software Roadmap 52, JetPack 4.6 was the last major release for Nano/TX1/TX2 and JetPack 5.x supports Xavier and Orin devices.
官方原话，意思是nano最高支持的jetpack版本就是4.6，也就是python3.6.9，torch最后一个支持3.6的版本就是1.10.0。如果你不是nano是xavier或者更高级的型号，那就选python3.8的版本。
安装常见问题
1.下载完轮子后用pip3安装，如果提示：
“error：torch is not a supported wheel on this platform”
如果你下载的是预编译好的，出现了这个问题，实际上是你的pip3版本和你的python3.6不兼容，我这边之前尝试更新python和pip的时候遗留下来了这种问题。
解决方法：
wget https://bootstrap.pypa.io/get-pip.py
sudo python3.6 get-pip.py
即重新安装对应版本的pip。
2.进入python3环境，尝试import torch时出现：
numpy.core.multiarray failed to import
或者
numpy illegal instruction (core dumped) jetson
意思就是你的numpy版本有问题，和torch不兼容！
解决方法：
首先卸载所有的numpy，再重新安装低版本numpy：
#1.19.5出错
pip3 install numpy==1.19.4
安装完毕的检验
输入以下命令查看：
import numpy
import torch
torch.__version__
torch.cuda.is_available()
如果没报异常且torch.cuda.is_available显示True，说明安装正常完成了，如下图所示：
