jetson nano 安装pytorch和pytorchvision

前言

随着人工智能的不断发展，机器学习这门技术也越来越重要,在嵌入式图像识别中我们难免会使用到jetson nano做图像处理，本章讲述怎么样在jetson nano上安装pytorch和pytorchvision两个重要的东西
安装需要的下载文件
下载链接：https://pan.baidu.com/s/1bQGfYtmOB-EaqvKnnJZE_w
提取码：lpeh

---

一、Jetson nano 的镜像系统

在英伟达官方网站可以找到烧写镜像的具体步骤，这里不在赘述，镜像烧写按照步骤一步一步来就可以了，内存卡推荐使用32gb以上的，镜像烧写完成后进入系统

二、修改镜像源

1.在终端里使用vim /etc/apt/sources.list将原来的内容注释掉换成中科大源（清华源下载比较慢）

# 默认注释了源码镜像以提高 apt update 速度，如有需要可自行取消注释
deb http://mirrors.ustc.edu.cn/ubuntu-ports bionic main restricted universe multiverse
# deb-src http://mirrors.ustc.edu.cn/ubuntu-ports bionic main restricted universe multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports bionic-updates main restricted universe multiverse
# deb-src http://mirrors.ustc.edu.cn/ubuntu-ports bionic-updates main restricted universe multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports bionic-backports main restricted universe multiverse
# deb-src http://mirrors.ustc.edu.cn/ubuntu-ports bionic-backports main restricted universe multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports bionic-security main restricted universe multiverse
# deb-src http://mirrors.ustc.edu.cn/ubuntu-ports bionic-security main restricted universe multiverse

2.系统更新：

sudo apt-get update
sudo apt-get upgrade -y

---

三.pytorch安装

1.首先安装pip和Cython

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

Cython如果下载速度慢可以在命令后面加临时软件源 -i https://pypi.mirrors.ustc.edu.cn/simple

pip3 install Cython -i https://pypi.mirrors.ustc.edu.cn/simple

下面开始安装pytorch，安装pytorch是可以用conda安装的，不过经过多次安装发现，conda需要配置虚拟环境有时python的版本设置可能不对，conda安装pytorch还是相当麻烦的，这里采用pytorch的离线包安装（下载链接前言中附赠）。
下载好文件对文件解压进入文件夹中

jetbot@jetbot-desktop:~/Downloads$ ls
pytorch-torchvison.zip
jetbot@jetbot-desktop:~/Downloads$ unzip pytorch-torchvison.zip
jetbot@jetbot-desktop:~/Downloads$ ls
pytorch-torchvison  pytorch-torchvison.zip
jetbot@jetbot-desktop:~/Downloads$ cd pytorch-torchvison/
jetbot@jetbot-desktop:~/Downloads/pytorch-torchvison$ ls
torch-1.8.0-cp36-cp36m-linux_aarch64.whl  torchvision

进入文件夹看到这俩，接着开始安装pytorch

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

等待安装结束进入python3中import验证一下pytorch的导包

四.安装pytorchvision

还是在当前目录下，进入torchvision文件夹里，在终端里键入以下命令对pytorchvision安装

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
sudo python3 setup.py install

等待pyutorchvision安装成功进入python3中import导包进行验证。