树莓派编译&安装pytorch教程

树莓派编译&安装pytorch教程

准备工作

1.系统进行更新

sudo apt update && sudo apt upgrade

2.安装vim（一个文本编辑器）

sudo apt install vim

3.增加交换区的大小到4G

使用vim打开dphys-swapfile文件
sudo vim /etc/dphys-swapfile

找到常量CONF_SWAPSEIZE，CONF_MAXSWAP，去掉前面#，值都改成4096（4G）

vim的使用方法可以参考我的另一篇博客：
vim使用教程

然后执行以下命令激活交换文件系统：

sudo /etc/init.d/dphys-swapfile stop
sudo /etc/init.d/dphys-swapfile start

4.创建虚拟环境

python3 -m venv pytorch
这时会在 home/pi 目录下生成一个新文件夹pytorch

如果要进入虚拟环境，可以执行：
source pytorch/bin/activate
这个时候我们可以在命令窗口的提示符前面见到(pytorch)

退出虚拟环境，可以执行：
deactivate

torch的编译和安装

1.从github下载源文件

mkdir pytorch_install  #在/home/pi目录下创建文件夹pytorch_install用于存放源码
cd pytorch_install
git clone --recursive https://github.com/pytorch/pytorch  #下载过程可能会有问题，我下载的时候mkl-dnn就没克隆下来,后面会补救

2.切换到我们想安装的版本：

cd pytorch
git checkout tags/v1.0.1 -b build
git submodule update --init --recursive  #这里会检索下载的源文件并重新下载缺失的文件

3.安装依赖包

编译之前要先安装依赖包，搭建环境

pip3 install -r requirements.txt   #安装numpy 等一些必要的包
sudo apt-get install libopenblas-dev cython3 libatlas-base-dev m4 libblas-dev cmake  #搭建环境

4.设置环境变量

比如：树莓派不支持cuda,所以需要设置环境变量，把cuda支持去掉，否则编译将会报错

export NO_CUDA=1
export NO_DISTRIBUTED=1
export NO_MKLDNN=1
export NO_NNPACK=1
export NO_QNNPACK=1

5.编译&安装

接着我们就可以利用命令：
python3 setup build
对pytorch进行编译，这个过程需要需要耗费几个小时
当编译完成后，直接安装
python setup install
这样pytorch就成功安装了

检测

进入虚拟环境，执行

python3
import torch
import numpy

没有报错就应该就可以了