[orin] nvidia orin 上安装 pytorch 和 torchvision 实操

请看这个博主写的链接，写的非常好，目前我已经安装成功了，不同的是我是在Anaconda虚拟环境中安装的。
原博客链接：https://blog.csdn.net/beautifulback/article/details/125717717

这次我要安装的软件版本是:python3.8 + torch1.11.0+ torchvision0.12.0

首先说明，我拿到这个orin设备之前，已经有很多人在上面配置了，所以可能前人已经安装了很多依赖。所以本次安装有可能存在有些依赖没有安装的情况。

安装过程

1.首先创建conda环境

conda create -n py38 -y

2.然后安装依赖：

sudo apt-get install  libopenblas-base libopenmpi-dev libomp-dev

3.安装Pytorch

# 先进入虚拟环境
conda activate py38
# 安装pip依赖
pip install Cython numpy
# 安装Pytorch
pip install  torch-1.11.0-cp38-cp38-linux_aarch64.whl

4.安装torchvision
解压压缩包vision，然后进入vision文件夹内，然后输入以下指令安装

export BUILD_VERSION=0.12.0  
python3 setup.py install --user

这一步会比较慢，耐心等待，如果有除了torchvision以外的包没有安装上，那么可以再额外进行安装即可，注意提示上的版本要求，例如下面的pillow版本要求不大于7。

安装pillow<7

pip install 'pillow<7'

5.查看版本

python

# 输入
import torch
import torchvision
torch.__version__
torchvision.__version__