树莓派3B安装pytorch1.0.0版本

树莓派3B安装pytorch1.0.0版本

经过三天的不断踩坑出坑，终于把pytorch1.0安装到了树莓派3B上

raspbian系统安装

首先树莓派的系统有内置的python版本，最新的raspbian系统中内置了python3.7，由于这个原因，导致很多基于python3.5的软件包安装不了，因此我找了一个内置python3.5的raspbian系统，链接如下：
https://pan.baidu.com/s/1DJreDT8sL47WcrfK6LWttQ 提取码：3pjc

一、系统源和pip源安装

系统源

1.将原系统的源改成中科大的源

sudo nano /etc/apt/sources.list
deb http://mirrors.ustc.edu.cn/raspbian/rapbian/ stretch main contrib non-free rpi

2.修改系统更新源

sudo nano /etc/apt/sources.list.d/raspi.list
deb http://mirrors.ustc.edu.cn/archive.raspberrypi.org/debian/ stretch main ui

3.更新软件源列表，软件版本和内核版本

sudo apt-get update

二、安装必要的软件包

sudo apt-get install libopenblas-dev cython3 libatlas-dev \m4 libblas-dev cmake
pip3 install numpy==1.11.1
pip3 install pyyaml

如果PIP安装失败出现红字，换个网就行，我直接用WIFI连手机热点就解决了

三、划分swap空间

1.划分空间

sudo apt-get install vim
sudo vi /etc/dphys-swapfile

然后将CONF_SWAPSIZE 和CONF_MAXSWAP的注释去掉，然后将SWAPSIZE的数字改为2048，然后wq退出。

2.使新设置的空间生效

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

四、下载pytorch官方的repository

1.下载

git clone https://github.com/pytorch/pytorch.git

2.选择分支

cd pytorch
git checkout tags/v1.0.0 -b build
git submodule update --init

3.下载编译好的wheel文件

网址: http://tomorrow.ai/shared/pytorch1.0_raspberry/torch-1.0.0a0+8322165-cp35-cp35m-linux_armv7l.whl
下载完成后复制到pytorch文件夹中
设置环境变量

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

安装编译好的whl文件

pip3 install torch-1.0.0(文件名)

五、测试pytorch

python3
import torch

完成