FPGA加速卷积神经网络(一)ubuntu中配置caffe环境

1.对Ubuntu进行一些调教

使用vmware新建一个ubuntu18.04.6的64位的Linux系统。
系统装好后会自动下载安装wmware tools,如果没有装,则需要手动安装。
接下来需要更换ubuntu的下载源,否则在后续安装模块的过程中会非常的满,甚至出现报错的情况。
点击桌面左下角的菜单栏,打开software&updates,在download from中选择清华的下载源。
在这里插入图片描述
完成后打开terminal执行下列命令更新软件

sudo apt-get update
sudo apt-get upgrade

2.caffe环境配置

下载所需要的依赖库

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev
sudo apt-get install libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

注意看执行过程中的输出信息,未能成功安装的库需要再次单独安装,否则后续在编译caffe的过程中会出现报错。

下载caffe

sudo git clone https://github.com/BVLC/caffe.git

在这里插入图片描述
配置Makefile.congfig文件
caffe文件夹中有 Makefile.config.example 文件,是 Caffe自带的 Makefile 的例子,复制这个文件并重命名为 Makefile.config。打开后删除如下选项的前的注释符号#。

CPU_ONLY := 1
OPENCV_VERSION := 3

编译caffe

make all
make test
make runtest

编译过程如果出现没有编译完成就停止编译的情况,可以使用 make clean之后重新编译
测试
完成编译后已经成功安装caffe,接下来使用cifar10进行测试。
在caffe目录下执行下列操作

./data/cifar10/get_cifar10.sh
./examples/cifar10/create_cifar10.sh

将 examples/cifar10/cifar10_quick_solver.prototxt 中的 GPU 改为 CPU
将examples/cifar10/cifar10_quick_solver_lr1.prototxt中的GPU改为CPU

./examples/cifar10/train_quick.sh

出现下面的画面则表示配置成功,可以手动停止。
在这里插入图片描述

3.pycaffe环境配置

Caffe 训练及测试时会用到 python 接口, 下面介绍 pycaffe 接口的配置。
下载相关库

sudo pip install --upgrade pip
sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python3-pip python-pip
sudo pip install python-skimage
sudo apt-get install python-h5py python-protobuf python-leveldb python-networkx pythonnose python-pandas python-gflags
sudo pip install Cython ipython

如果有一些库下载失败,可以先完成其他库的安装,最后再来重新安装下载失败的库。
编译python接口
进入caffe目录,执行编译

make pycaffe

添加环境变量
在/etc/profile 文件的最后一行中添加环境变量:

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

然后在source一下让变量生效。

source /etc/profile

测试
进入caffe下的python目录

python
import caffe

如果没有出现报错则python接口配置成功。
注意这里的python版本使用的是ubuntu自带的2.7,不要使用3以上的版本。否则会出现import失败的情况。
如果在2.7版本下import失败可以尝试make clean后再重新编译。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值