Nvidia Jetson篇----Jetson Xavier nx安装pytorch

1. 安装相关依赖

sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev
sudo apt-get install liblapack-dev libblas-dev gfortran

sudo apt-get install libopenblas-base libopenmpi-dev
sudo pip3 install mpi4py
sudo pip3 install Cython

2. 安装pytorch

下载pytorch编译好的.whl文件
下载地址：https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-6-0-now-available/72048

sudo pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl 

3. 安装torchvision

直接pip3下载是没有torchvision0.7版本，需要 git 下载，然后编译安装
打开git，网址：https://github.com/pytorch/vision
找到 torch 对应的 torchvision 版本

sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch  v0.9.0 https://github.com/pytorch/vision torchvision

cd进入到目录中

 cd torchvision
 sudo python3 setup.py install