Nvidia jetson xavier agx 安装pytorch1.9.0 Gpu版

        最近在学深度学习，手头笔记本都不是英伟达显卡，只能用cpu硬抗，真的是慢的怀疑人生。不过手头有一个Nvidia jetson xavier agx官方开发套件，性能可观，就想拿这个训练模型。

        刚开始以为安装方式和正常Ubuntu环境一样，其实这样也可以安装，不过这样安装的是pytorch cpu版本的，cuda用不了，想用cuda还是得装gpu版本，agx  pytorch_gpu只能通过英伟达官方发布的包安装，以下就记录一下详细安装过程。

一、agx内置环境

1、jetpack4.6.1

2、cuda10.2

3、python3.6

二、torch安装

        常规的安装基本都是基于anaconda3进行的，但是agx好像不用anaconda，我用anaconda安装一直报错，用不了。（要是谁装成功了，指导我一下。）

1、文件下载

        可通过Jetson Zoo - eLinux.org网站自行下载，不过需要翻墙。

我下载的是torch1.9.0 版本，以下是百度网盘链接。

链接：https://pan.baidu.com/s/1Mdm_Q7G21pIdbUmkoXHbUQ?pwd=u7m2 
提取码：u7m2 

2、安装

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev

sudo pip3 install Cython 

pip3 install torch-1.9.0-cp36-cp36m-linux_aarch64.whl

测试一下，cuda可以用了。

三、安装torchvision

        torchvision也不能通过常规方式进行安装，torch1.9对应torchvision0.10，以下给出安装方式。
 

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

git clone --branch release/0.10 https://github.com/pytorch/vision torchvision 

cd torchvision export BUILD_VERSION=0.10.0

sudo python3 setup.py install --user

安装完成,你以为真的完了，想多了，哈哈哈，大概率会报错，如下图。

不过不要慌，找到问题就能解决，也很简单，更新一下pillow就ok了。

 pip3 install --upgrade pillow -i https://pypi.tuna.tsinghua.edu.cn/simple

以上就是在英伟达 agx 上安装GPU 版本的pytorch，踩坑无数啊。

要是有更好的安装方式，大家可以分享给我，同时欢迎交流指正，私信我就行了。