jetson nano 安装pytorch

这里默认你已经装好了jetson nano的系统镜像了

更换apt源
Jetson设备开机就要更换国内的源，否则基本啥事都干不了。

这里用的是中科大的源，内容如下：

sudo mv /etc/apt/sources.list /etc/apt/sources.list.bak # 将原镜像源文件做一个备份
sudo gedit /etc/apt/sources.list # 用文本编辑器打开源文件进行编辑，将下面中科大的apt源内容写入这个文件中。命令行模式就用vi吧

中科大镜像源

deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates main restricted
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security main restricted
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security multiverse

将文件保存之后，推荐执行一次系统全面升级：
sudo apt update
sudo apt full-upgrade

系统升级之后，需要先安装相关的依赖库和工具，至少pip3 install要能用是吧~

sudo apt-get install libhdf5-serial-dev hdf5-tools zlib1g-dev zip libjpeg8-dev libhdf5-dev  python3-pip

pip3换国内镜像源

cd ~
mkdir .pip # 创建文件夹
nano .pip/pip.conf # 将配置内容写入配置文件即可

配置内容如下：

[global]
trusted-host = mirrors.aliyun.com
index-url = http://mirrors.aliyun.com/pypi/simple

安装相关的python库

pip3 install cython #numpy编译需要用到
pip3 install -U numpy  # 需要编译安装，用时很长，所以单独安装
pip3 install -U h5py # 需要编译安装，用时非常长，我的板子装这个20多分钟才装完
pip3 install -U grpcio absl-py py-cpuinfo psutil portpicker six mock requests gast astor termcolor

安装pytorch
jetson nano可以用官方适配的whl安装，自行编译不仅耗费时间，还会出错。官方地址，下载pytorch可能需要

下载完成后直接pip3 install安装即可，这时进python后import torch可能会出现缺失相关so文件的问题

需要到这个网站右上角搜索缺失的库文件，如我这里就是libmpi_cxx.so.20，然后下载arm64版的

点击进去后到download板块寻找下载地址下载

下载好后安装到jetson nano中，如果安装过程又提示相关缺失库，就使用apt install安装或者照着之前的操作去上面网址搜索下载安装

安装torchvsion
成功安装完torch文件还需要安装torchvsion，pytroch对应的vision如下图所示：

下载直接到官网下载，在下图位置搜索对应版本

下载完后解压到nano中

cd torchvision
export BUILD_VERSION=x.x.x  #x.x.x是你的版本，如我的pytroch1.7对应为0.8.1
python3 setup.py install --user

到此位置，整个pytroch安装过程就可以了