Pytorch是著名的人工智能框架，在部署人工智能项目的时候，就需要在嵌入式开发板上安装pytorch，下面就详解介绍关于安装pytorch的步骤
1.更换apt源
首先检查apt源是否已经是国内镜像源
输入：
sudo apt-get update
如果不是国内镜像的话
依次输入一下命令
cd /etc/apt
cp /etc/apt/sources.list /etc/apt/sources.list.bak
gedit sources.list
打开之后把里面的内容替换为;
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main multiverse restricted universe
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main multiverse restricted universe
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main multiverse restricted universe
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main multiverse restricted universe
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main multiverse restricted universe
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main multiverse restricted universe
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main multiverse restricted universe
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main multiverse restricted universe
再次输入
sudo apt-get update
查看是否换源成功
2.更换pip源
输入：
pip3 install --upgrade pip
检查是否换源
如果没有依次输入以下命令
mkdir ~/.pip
gedit ~/.pip/pip.conf
把一下内容粘贴进去：
[global]
index-url = https://pypi.mirrors.ustc.edu.cn/simple
[install]
trusted-host = https://pypi.mirrors.ustc.edu.cn
既可以完成pip换源
3.开始安装pytorch
这里以torch-1.8.0和torchvison0.9.0为例
一定一定要把torch和torchvison版本对应起来
具体版本如何对应请查看
开始安装torch依赖
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython
安装pytorch
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
安装完之后运行python3，输入
import torch
print(torch.__version__)
查看是否安装成功
安装torchvison
输入：
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
cd torchvision
export BUILD_VERSION=0.9.0 #python3 setup.py install --user
cd ../
pip install 'pillow<7'
启动python3输入：
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)
检查完全安装成功