树莓派编译安装caffe2

平心而论,树莓派平台可以选择的深度学习框架还是相当多的,这里主要记录下caffe2平台的安装。

首先按照官网教程,clone源码:

git clone --recursive https://github.com/pytorch/pytorch.git
git submodule update --init

考虑到树莓派可能不是很稳定,用ubuntu系统clone下来打包丢到树莓派里边也许会好些。。。

之后由于树莓派内存限制,我们要先把树莓派的交换区扩大到2G。输入:

sudo nano /etc/dphys-swapfile

将CONF_SWAPSIZE=100改为=2048以上,保存退出。之后用下列命令激活交换区:

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

之后由于caffe2合并到pytorch了,所以应该用下列指令编译,而不是如官网所述cd caffe2:

cd pytorch
./scripts/build_raspbian.sh

之后根据报错的提示,pip安装相应软件包,我这边安装了pyyaml,不过我的树莓派在之前安装了Tengine平台,不清楚会不会附带安装了一些依赖插件,等不报错开始正常编译之后就是等了,大概3~4小时吧,编译完成之后,就可以吧交换区改回100了,毕竟设置的太大对TF卡不好。

编译完成之后安装:

cd build
sudo make install

现在打开python(注意不是python3),输入import caffe2,不报错说明这一步安装成功。

以下操作出自https://www.jianshu.com/p/c5067b814f9a

输入import caffe2.python.predictor.predictor_exporter as pe,正常情况下会报错,因为python中protobuf的版本和系统版本不一致,可通过下列指令查看:

pkg-config --modversion protobuf
pip list | grep protobuf

不一致没关系,我们执行

sudo pip install --upgrade protobuf==3.6.1 #根据你系统的版本改

没安装future库的要装上,不然报错

sudo pip install future

更新protobuf失败可能是pip版本过低,需要执行:

pip install --upgrade pip
sudo pip install --upgrade pip

好啦,现在caffe2就编译安装完毕了。

你可能感兴趣的:(神经网络)