caffe2在树莓派(raspberry 3)中试用

来源:互联网 发布:虚拟机mac显卡驱动 编辑:程序博客网 时间:2024/05/21 09:06

在树莓派中安装caffe2并且试跑了mnist数据。

本身对树莓派不是很熟,加上build速度太慢,折腾了一晚上。

caffe2在树莓派3中的安装有出现什么大的问题,就是在按着教程:MNIST - Create a CNN from Scratch,试着跑mnist数据集时出现了一点问题。还有需要注意,在树莓派中使用matplotlib 等画图的包让树莓派太费劲了,因此,不使用这些可视化的包。

在转化mnist数据到eveldb出现了:

*** SIGSEGV (@0x0) received by PID 14864 (TID 0x76ff2000) from PID 0; stack trace: ***@ 0x765ee1a0 (unknown)@ 0x137a0 caffe2::convert_dataset()@ 0x12b0c main@ 0x765d7294 (unknown)Segmentation fault

在#376中也有提到,有人提出了一些方法,可以试一下。我解决的方法为:

  • 安装libleveldb-dev, libsnappy-dev,(其中可能出现版本冲突等问题,可以降低版本解决)
  • 重新编译caffe2,并且注意cmake配置时LEVELDB开关为ON。

通过重新编译,mnist数据的转换就没有问题了,且训练结果如下:

1

原创粉丝点击