NX配置pytorch，torchvision环境

1. 系统版本情况

NX系统：Ubuntu18.04
python版本：3.6.9
JetPack：4.4
cuda版本：10.2
安装的Pytorch版本为1.6.0，torch vision版本为0.7.0.

2. 安装pytorch

推荐网上下载whl安装包安装，安装包下载地址在这里，因为NX是arm架构的，所以这里下载aarch64的安装包，如：Python 3.6 - torch-1.6.0-cp36-cp36m-linux_aarch64.whl下载完成后，执行

pip3 install XXX.whl

XXX.whl为下载的安装包名称。
安装完成后，执行一下指令测试：

python3
import torch

如果不报错，则安装成功。

3. 安装torchvision

安装好pytorch后，选择合适版本的torchvision安装，pytorch和torchvision有严格的对应关系，这里要注意，这里给出各个版本的对应关系。
因为torchvision不像torch那样有安装包可以下载下来本地安装，找不到torchvision aarch64架构的安装包，所以这里需要下载源码手动编译安装才行。
torchvision源码地址。
安装指令：

git clone --branch v0.7.0 https://github.com/pytorch/vision torchvision 
cd torchvision
export BUILD_VERSION=0.7.0 
sudo python setup.py install 

这里一定记得加sudo。
如果cuda版本正确的话，一般编译不会有错误。编译完成后，在python3中import测试。不报错则安装成功。

有问题可以留言哦。