Jetson xavier nx 入门系列——pytorch环境配置

以下安装过程同样适用于jetson agx xavier

一、安装依赖

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

二、安装python环境以及依赖包

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install git cmake python3-dev
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev

配置python环境

 sudo apt-get install python3-pip
 sudo pip3 install -U pip testresources setuptools

三、安装pytroch

nx板刷机系统为JP4.4版，自带cuda版本为10.2，所以必须pytorch1.5版本以上才可以，系统自带cudnn，这些无需配置。
下载pytorch编译好的.whl文件（1.6版本）

链接：https://pan.baidu.com/s/1-ZGABzludm4t1l6z9s9b1Q 
提取码：rc51 

下载后，将文件上传NX板，进行如下操作：

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

三、安装torchvision

此处安装torchvision v0.7.0版本

安装依赖

sudo apt-get install libjpeg-dev zlib1g-dev
sudo apt install libavcodec-dev
sudo apt install libavformat-dev
sudo apt install libswscale-dev

libavcodec-dev、libavformat-dev、libswscale-dev这三个包有时不安装会报错，填坑参考链接

下载torchvision

 git clone --branch  v0.7.0 https://github.com/pytorch/vision torchvision

以上可能会下载错误，可以使用码云将https://github.com/pytorch/vision导入，然后进行下载，以下是我自己导入的仓库，可以直接使用：

 git clone --branch  v0.7.0 https://gitee.com/tulipdi/vision.git torchvision

进入下载好的目录，进行安装

 cd torchvision
 sudo python3 setup.py install

安装到最后需要pillow包，最后可能会下载失败。但是torchvision已经安装成功了，输入命令pip3 list可以看到。
然后pip3 install pillow安装pillow包即可，如果下载速度慢，可以加入-i加清华源下载。

至此安装完成

参考文献：
NVIDIA Jetson NX刷机安装pytorch（看这一篇就够了！！！）