码迷,mamicode.com
首页 > 系统相关 > 详细

安装配置ubuntu14.04+caffe

时间:2016-06-26 16:37:02      阅读:195      评论:0      收藏:0      [点我收藏+]

标签:

了解caffe安装,记录安装过程。参考资料:

1.caffe官网 http://caffe.berkeleyvision.org/

2.caffe的Git教程 https://github.com/BVLC/caffe

3. denny学习笔记 http://www.cnblogs.com/denny402/p/5067265.html

4. caffe环境搭建 http://blog.csdn.net/hjimce/article/details/48781693

5. http://blog.csdn.net/liuxiabing150/article/details/42503389

 

一、版本

linux系统:Ubuntu 14.04 (64位)

预留足够内存,否则会出现内存溢出的问题。之前虚拟机只留了2G内存,明显不够用,又调整了内存大小重新配置。

 

二、开始安装

 1、因个人电脑显卡不满足运算,省去了安装CUDA,安装无GPU版本。

 2、安装caffe

   先下载caffe

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

然后安装一堆第三方库

技术分享
# sudo apt-get install libatlas-base-dev
# sudo apt-get install libprotobuf-dev
# sudo apt-get install libleveldb-dev
# sudo apt-get install libsnappy-dev
# sudo apt-get install libopencv-dev
# sudo apt-get install libboost-all-dev
# sudo apt-get install libhdf5-serial-dev
# sudo apt-get install libgflags-dev
# sudo apt-get install libgoogle-glog-dev
# sudo apt-get install liblmdb-dev
# sudo apt-get install protobuf-compiler
技术分享

接着,安装opencv。第一次安装遇到内存溢出问题,后扩展内存后,避免了这个问题

# cd caffe
# sudo git clone https://github.com/jayrambhia/Install-OpenCV
# cd Install-OpenCV/Ubuntu
# sudo sh dependencies.sh
# cd 2.4
# sudo sh opencv2_4_10.sh

接下来,编译caffe。

# cd caffe
# sudo cp Makefile.config.example Makefile.config

因为是安装无GPU版,所以需要修改Makefile.config。文件内容如下:

技术分享

然后我们把:#CPU_ONLY:=1,那一行的注释符号去掉:CPU_ONLY:=1。修改时,执行以下语句:

 

# sudo vi /etc/Makefile.config

 

 注意vi命令的使用方法,(:wq)保存退出。

 

完成上述设置后,开始编译: 

 #   make all -j4  
 #   make test  
 #   make runtest 

注意:-j4 是指使用几个线程来同时编译,可以加快速度,j后面的数字可以根据CPU core的个数来决定,如果CPU是4核的,则参数为-j4,也可以不添加这个参数,直接使用“make all”,这样速度可能会慢一点儿。 

编译时遇到错误“caffe/proto/caffe.pb.h: No such file or directory”,执行以下语句:

# protoc src/caffe/proto/caffe.proto --cpp_out=.  
# mkdir include/caffe/proto  
# mv src/caffe/proto/caffe.pb.h include/caffe/proto 

 至此,caffe安装完成。

3、测试caffe

  下载mnist数据

# cd caffe
# sudo sh data/mnist/get_mnist.sh
# sudo sh examples/mnist/create_mnist.sh

运行时,没有GPU,则需要修改配置文件lenet_solver.prototxt

# sudo vi examples/mnist/lenet_solver.prototxt

将最后一行的solver_mode:GPU改为solver_mode:CPU

配置好后,就可以运行了

# sudo sh examples/mnist/train_lenet.sh

注意,运行caffe程序时,必须在caffe的根目录下,不然会出错。

安装配置ubuntu14.04+caffe

标签:

原文地址:http://www.cnblogs.com/Bunnyxiaochouchou/p/5618022.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!