原文地址：http://blog.csdn.net/hjimce/article/details/48781693

作者：hjimce

對(duì)于caffe的系統(tǒng)一般使用linux系統(tǒng)，當(dāng)然也有windows版本的caffe，不過(guò)如果你一開(kāi)始使用了windows下面的caffe，后面學(xué)習(xí)的過(guò)程中，會(huì)經(jīng)常遇到各種錯(cuò)誤，網(wǎng)上下載的一些源碼、模型也往往不能快速的跑起來(lái)，因?yàn)槊菜芻affe的官方只提供了linux版本，而且caffe在不斷的快速迭代更新中，如果不使用原版的話，后面編譯出現(xiàn)什么問(wèn)題，自己怎么錯(cuò)的，自己都不知道。本篇博文主要講解快速搭建caffe環(huán)境：

電腦系統(tǒng)：ubuntu 14.04

顯卡：GTX 850

在ubuntu下要完整的搭建caffe，個(gè)人感覺(jué)最難的一步就是cuda的安裝了，特別是對(duì)于雙顯卡的電腦來(lái)說(shuō)，很容易黑屏、無(wú)法登陸圖形界面，這個(gè)我安裝了n久，都沒(méi)裝成功，因?yàn)槲业碾娔X筆記本雙顯卡，每次裝完cuda就黑屏，網(wǎng)上的教程一大堆，但都中看不中用，導(dǎo)致我重裝了二三十次的系統(tǒng)，最后才成功。這里為了講caffe的安裝，我們先不使用GPU，進(jìn)行安裝測(cè)試，因?yàn)闆](méi)有GPU我們依舊可以跑caffe，只是速度比較慢而已。

1、安裝caffe所需要的依賴(lài)庫(kù)

命令：

sudo apt-get install libatlas-base-dev

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

這些庫(kù)要安裝挺久的，請(qǐng)耐心等待。

2、下載caffe。

到github上下載caffe：https://github.com/BVLC/caffe。下載完成后，解壓caffe源碼包。解壓后，我們打開(kāi)文件，可以看到caffe的源碼包如下：

3、配置Make.config 文件。caffe文件解壓后，文件夾下面有一個(gè)Makefile.config.example文件，我們需要對(duì)這個(gè)文件進(jìn)行重命名為：Make.config ?。也就是去掉后綴example。然后我們打開(kāi)這個(gè)文件，可以看到如下內(nèi)容：

然后我們把：#CPU_ONLY:=1，那一行的注釋符號(hào)去掉：CPU_ONLY:=1。這是因?yàn)槲覀儧](méi)有安裝CUDA，還不能使用gpu，所以我們把配置改為只使用cpu。

4、編譯caffe。

方案一:

(1)在完成Make.config配置后，我們輸入命令：

make all

進(jìn)行caffe源碼編譯.這一步有可能遇到如下錯(cuò)誤:

caffe/proto/caffe.pb.h: No such file or directory

如果出現(xiàn)這個(gè)錯(cuò)誤,那么輸入命令:

protoc src/caffe/proto/caffe.proto --cpp_out=.

mkdir include/caffe/proto

mv src/caffe/proto/caffe.pb.h include/caffe/proto

然后在進(jìn)行make all 就可以了

(2)編譯完成后，在安裝python接口，輸入命令：

make pycaffe

這個(gè)如果不使用python接口，調(diào)用caffe模型的話也可以不用安裝，不過(guò)建議還是搞一下，就一句話的事。完事后，我們會(huì)發(fā)現(xiàn)caffe源碼目錄下，多了一個(gè)build文件，這個(gè)文件下面有個(gè)tools，打開(kāi)這個(gè)文件夾：

這個(gè)文件夾下面的工具可是個(gè)好東西啊，以后我們會(huì)經(jīng)常用到這些可執(zhí)行文件，最常用的就是可執(zhí)行文件：caffe，我們只要調(diào)用這個(gè)工具，就可以進(jìn)行訓(xùn)練。

(3)接著編譯test文件夾下面的源碼。命令如下：

make test

make runtest

采用這種方案一般沒(méi)問(wèn)題,不過(guò)我在使用c++調(diào)用的時(shí)候,會(huì)使用到鏈接庫(kù):libcaffe.so.1.0.0-rc3,這種方法編譯后沒(méi)有生成這個(gè)文件;經(jīng)過(guò)google查找,發(fā)現(xiàn)采用cmake編譯,才會(huì)生成libcaffe.so文件

方案二:直接采用cmake:

mkdir build

cd build

cmake ..

make all -j8

5、測(cè)試階段

安裝完了，自然要測(cè)試一下能不能用咯。首先cd到caffe目錄，然后輸入命令：

sh data/mnist/get_mnist.sh

sh examples/mnist/create_mnist.sh

vim examples/mnist/lenet_solver.prototxt

把lenet_solver.prototxt里面的solver_mode 改為 CPU。因?yàn)槲覀冞€沒(méi)裝GPU，暫時(shí)只使用CPU就好了。

然后我們運(yùn)行腳本：

./examples/mnist/train_lenet.sh

這個(gè)時(shí)候，如果成功的話，就會(huì)開(kāi)始跑起來(lái)：

說(shuō)明：如果在使用caffe、或者編譯安裝caffe過(guò)程中，出現(xiàn)如下錯(cuò)誤：

CXX/LD -o .build_release/tools/convert_imageset.bin

.build_release/lib/libcaffe.so: undefined reference tocv::imread(cv::String const&, int)'.build_release/lib/libcaffe.so: undefined reference tocv::imencode(cv::String const&, cv::_InputArray const&, std::vector >&, std::vector > const&)'

那么請(qǐng)修改上面的Makefile文件（不是Makefile.config）：

LIBRARIES += glog gflags protobuf leveldb snappy \

lmdb boost_system hdf5_hl hdf5 m \

opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs

也就是在libraries后面，加上opencv的相關(guān)庫(kù)文件。

接著就開(kāi)始caffe搞起吧，推薦個(gè)caffe模型網(wǎng)站：https://github.com/BVLC/caffe/wiki/Model-Zoo。本來(lái)個(gè)人不是很喜歡caffe的，就是因?yàn)檫@個(gè)網(wǎng)站吸引了我，這個(gè)網(wǎng)站可以搞到好多caffe模型、源碼，非常適合于我們學(xué)習(xí)。

二、在Eclipse中使用編譯調(diào)試caffe

1、首先就是安裝Eclipse，然后安裝c++開(kāi)發(fā)插件，這個(gè)可以百度搜一下，eclipse下面怎么進(jìn)行c++開(kāi)發(fā)。

2.導(dǎo)入caffe makefile工程到eclipse （由于是英文版，下面描述也用英文，省的翻譯，方便大家調(diào)試）
（1）File→New→Project→C/C++ →Makefile Project with Existing Code.
（2）Create a new Makefile Project from existing code
Projectname:?caffe-master
Existing code location：/home/user/caffe-workspace/caffe-master
Language:?choose C and C++
Toolchain:choose Linux GCC
（3）Then click on caffe-master in Project Explorer (set Window→Open ?Perspective → C/C++).

（4）Now go File → Properties → Run/Debug settings.Click ?New.., and choose C/C++ application
（5）Fill launch configurationproperties
· ? ? ? ?Arguments:
fill ? train –solver=examples/mnist/lenet_solver.prototxt
and change working directory from default to /home/user/caffe-workspace/caffe-master（change to your own directory）

(6)Now you can use debug caffe code: Run-> Debug

三、C++函數(shù)調(diào)用相關(guān)路徑，makefile

CC=g++

CXXFLAGS = -O2 -Wall -D__STDC_CONSTANT_MACROS

INCLUDE = -I/usr/local/cuda/include -I. -I/usr/local/cuda/include -I/home/hjimce/caffe/include/ -I/home/hjimce/caffe/src/

LIBRARY = -L/usr/local/x86_64-linux-gnu/ -lprotobuf \

-L/usr/lib/x86_64-linux-gnu/ -lglog \

-L/usr/local/cuda/lib64/ -lcudart -lcublas -lcurand \

-L/usr/local/lib/ -lm -lpthread -lavutil -lavformat -lavcodec -lswscale -lopencv_core -lopencv_imgproc -lopencv_highgui \

-L/usr/lib/python2.7/config-x86_64-linux-gnu/ -lpython2.7 \

-L/sur/lib32/ -lrt \

-L../../caffe/build/lib/ -lcaffe

all:

$(CC) $(INCLUDE) $(OBJS) testcpp.cpp -o exercise $(LIBRARY)

windows下的caffe環(huán)境搭建

最近在ubuntu搞了一個(gè)月的caffe，總感覺(jué)很不爽，因?yàn)閡buntn下面的c++集成開(kāi)發(fā)工具，eclipse用起來(lái)沒(méi)有vs爽，因?yàn)閷?duì)caffe的函數(shù)名不是很熟悉，所以需要借助vs的c++助手。然后前一個(gè)月大部分也是調(diào)用pycaffe，但是最近感覺(jué)需要對(duì)caffe的c++函數(shù)比較熟悉，才能把自己的能力進(jìn)一步提高，于是就開(kāi)始搞起了windows 下的caffe，借助vs的強(qiáng)大功能，快速學(xué)習(xí)caffe。一開(kāi)始采用vs2012，最后各種錯(cuò)誤，最后改成vs2013很容易就編譯成功了。

最后推薦一個(gè)windows下caffe安裝網(wǎng)站：http://thirdeyesqueegee.com/deepdream/2015/07/13/running-deep-dream-on-windows-with-full-cuda-acceleration/? ? 這個(gè)包含了python的集成開(kāi)發(fā)環(huán)境anaconda，caffe的全套功能。

************作者：hjimce ? 時(shí)間：2015.10.10 ?聯(lián)系QQ：1393852684 ?原創(chuàng)文章，轉(zhuǎn)載請(qǐng)保留原文地址、作者等信息***************

附錄:在使用pycaffe遇到如下錯(cuò)誤

raise ValueError('Mean shape incompatible with input shape.')

ValueError: Mean shape incompatible with input shape.

那么把??./python/caffe/io.py文件夾下面的代碼:

if ms != self.inputs[in_][1:]:

raise ValueError('Mean shape incompatible with input shape.')

改成:

if ms != self.inputs[in_][1:]:

print(self.inputs[in_])

in_shape = self.inputs[in_][1:]

m_min, m_max = mean.min(), mean.max()

normal_mean = (mean - m_min) / (m_max - m_min)

mean = resize_image(normal_mean.transpose((1,2,0)),

in_shape[1:]).transpose((2,0,1)) * \

(m_max - m_min) + m_min

#raise ValueError('Mean shape incompatible with input shape.')

創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來(lái)咯，堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)

總結(jié)

以上是生活随笔為你收集整理的ubuntu 系统下的Caffe环境搭建的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。