在nx上安装torch，先下载torch安装包。
wget https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl -O torch-1.6.0-cp36-cp36m-linux_aarch64.whl
安装torch命令：
sudo pip3 install Cython #防止安装torch报错
pip3 install torch-1.6.0-cp36-cp36m-linux_aarch64.whl
安装torchvision:
先安装依赖：
sudo apt-get install libjpeg-dev zlib1g-dev libopenblas-dev
下载安装包：
git clone --branch v0.7.0 https://github.com/pytorch/vision torchvision
安装步骤：
cd torchvision
export BUILD_VERSION=0.7.0
sudo python3 setup.py install
这个时候报错：
fatal error: libavcodec/avcodec.h: No such file or directory
#include <libavcodec/avcodec.h>
我只需要把ffmpeg屏蔽掉就可以;修改setup.py
把 if has ffmpeg 修改为 if False
