NVIDIA TX2 安裝pytorch1.8

1、下载TX2专用torch包（.whl）不能去pytorch的官网找安装指令。
TX2专用下载地址，要和jetpack的版本对应好（网址里也有很清晰的安装教程）pytorch.whl下載
2、配备环境以及安装

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip install Cython
pip install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

安装慢可添加国内镜像源

清华：https://pypi.tuna.tsinghua.edu.cn/simple
阿里云：http://mirrors.aliyun.com/pypi/simple/
中国科技大学：https://pypi.mirrors.ustc.edu.cn/simple/
华中理工大学：http://pypi.hustunique.com/
山东理工大学：http://pypi.sdutlinux.org/
豆瓣：http://pypi.douban.com/simple/

注意：python版本必须和pytorch版本对应，否则安装不上；python3.6对应的是pytorch1.8

检测pytorch是否安装成功：

python
>>>import torch
>>>torch.cuda.is_available()
>>>True

若出现：

>>> import torch 
Illegal instruction (core dumped)

问题解决
在网上寻找解决方案时，看到了这个网页：https://stackoverflow.com/questions/65631801/illegal-instructioncore-dumped-error-on-jetson-nano

解决的方法就是增加：OPENBLAS_CORETYPE=ARMV8
可以使用临时添加方法，在运行Python指令前运行：
export OPENBLAS_CORETYPE=ARMV8

也可以采用增加系统变量方法，可以进行全局修改。
将“export OPENBLAS_CORETYPE=ARMV8”加入到“~/.bashrc”中
想弄清楚原因，可以查阅这个：
https://www.reddit.com/r/JetsonNano/comments/ktcyoh/illegal_instructioncore_dumped_error_on_jetson/

有时候会报错
OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory
解決辦法是需要安裝一個包
dpkg -i libopenmpi2_2.1.1-8_arm64.deb
参考链接
ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory

3、安装 torchvision

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
pip3 install setuptools #这个官方教程没写
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision 
cd torchvision


sudo python3 setup.py install

sudo python3 setup.py install若失败，去掉sudo再执行
4、测试

python3
>>>import torch
>>>print(torch.cuda.is_available()) 

出现true一般就可以了，需要其他的库就使用pip3