Ubuntu16.04配置安裝caffe

電腦配置
系統(tǒng):Ubuntu16.04
GPU:NVIDIA GTX1080

安裝依賴

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install -y libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install -y --no-install-recommends libboost-all-dev
# (Python general)
sudo apt-get install -y python-pip
# (Python 2.7 development files)
sudo apt-get install -y python-dev
sudo apt-get install -y python-numpy python-scipy

安裝Caffe
(1) 從github上獲取caffe (get git: sudo apt-get install git):

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

在caffe-master/python文件夾內,使用root執(zhí)行依賴項的檢查與安裝:

sudo su
cd caffe-master/python
for req in $(cat requirements.txt); do pip install $req; done

以上過程報錯寞缝,需要解決后再進行丈钙,可通過升級pip解決部分問題:

pip --version
pip install -u pip

(2) cp Makefile.config:

sudo cp Makefile.config.example Makefile.config

(3) 打開并修改配置文件:

sudo gedit Makefile.config #打開Makefile.config文件

3.1 使用CUDNN:

將
#USE_CUDNN := 1
修改成: 
USE_CUDNN := 1

3.2 使用OpenCV

USE_OPENCV := 1 (可選)
OPENCV_VERSION := 3 (如果打開了USE_OPENCV,而且opencv版本為3時)

3.3 Python來編寫layer

將
#WITH_PYTHON_LAYER := 1 
修改為 
WITH_PYTHON_LAYER := 1

3.4 路徑變化

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 
修改為: 
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial 

修改makefile文件
打開makefile文件蛉腌,做如下修改:

將:
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
替換為:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

編譯

make all -j8  -數(shù)字6為并行編譯進程數(shù)目榜掌,一般設置為CPU物理核心數(shù)溶锭,可以使用 $(nproc) 來替換或echo $(nproc) 來查看CPU物理核心數(shù)
make test -j8
make runtest -j8
make pycaffe  -此時應該已經編譯完成宝恶,再次執(zhí)行以確認

忽視以下錯誤即可:

nvcc warning : The 'compute_20', 'sm_20', and 'sm_21' architectures are deprecated, and may be removed in a future release (Use -Wno-deprecated-gpu-targets to suppress warning).

測試
MNIST數(shù)據集測試
配置caffe完成后,我們可以利用MNIST數(shù)據集對caffe進行測試,過程如下:
1.將終端定位到Caffe根目錄

cd ~/caffe

2.下載MNIST數(shù)據庫并解壓縮

./data/mnist/get_mnist.sh

3.將其轉換成Lmdb數(shù)據庫格式

./examples/mnist/create_mnist.sh

4.訓練網絡

./examples/mnist/train_lenet.sh

訓練開始:

I0922 13:14:47.332710  1418 caffe.cpp:218] Using GPUs 0
I0922 13:14:47.390920  1418 caffe.cpp:223] GPU 0: TITAN X (Pascal)
I0922 13:14:47.593374  1418 solver.cpp:44] Initializing solver from parameters: 
test_iter: 100
test_interval: 500
base_lr: 0.01
display: 100
max_iter: 10000
lr_policy: "inv"
gamma: 0.0001
power: 0.75
momentum: 0.9
weight_decay: 0.0005
snapshot: 5000
snapshot_prefix: "examples/mnist/lenet"
solver_mode: GPU
device_id: 0
...
I0922 13:15:00.678990  1418 solver.cpp:447] Snapshotting to binary proto file examples/mnist/lenet_iter_10000.caffemodel
I0922 13:15:00.682633  1418 sgd_solver.cpp:273] Snapshotting solver state to binary proto file examples/mnist/lenet_iter_10000.solverstate
I0922 13:15:00.684578  1418 solver.cpp:310] Iteration 10000, loss = 0.00334544
I0922 13:15:00.684592  1418 solver.cpp:330] Iteration 10000, Testing net (#0)
I0922 13:15:00.732586  1429 data_layer.cpp:73] Restarting data prefetching from start.
I0922 13:15:00.733906  1418 solver.cpp:397]     Test net output #0: accuracy = 0.9908
I0922 13:15:00.733917  1418 solver.cpp:397]     Test net output #1: loss = 0.0276493 (* 1 = 0.0276493 loss)
I0922 13:15:00.733922  1418 solver.cpp:315] Optimization Done.
I0922 13:15:00.733924  1418 caffe.cpp:259] Optimization Done.

參考:
http://blog.csdn.net/aifei7320/article/details/73252635(此文記錄了部分錯誤記錄)
http://www.reibang.com/p/7df78120803a

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末垫毙,一起剝皮案震驚了整個濱河市霹疫,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌综芥,老刑警劉巖更米,帶你破解...
    沈念sama閱讀 219,490評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異毫痕,居然都是意外死亡征峦,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,581評論 3 395
  • 文/潘曉璐 我一進店門消请,熙熙樓的掌柜王于貴愁眉苦臉地迎上來栏笆,“玉大人,你說我怎么就攤上這事臊泰◎燃樱” “怎么了?”我有些...
    開封第一講書人閱讀 165,830評論 0 356
  • 文/不壞的土叔 我叫張陵缸逃,是天一觀的道長针饥。 經常有香客問我,道長需频,這世上最難降的妖魔是什么丁眼? 我笑而不...
    開封第一講書人閱讀 58,957評論 1 295
  • 正文 為了忘掉前任,我火速辦了婚禮昭殉,結果婚禮上苞七,老公的妹妹穿的比我還像新娘。我一直安慰自己挪丢,他們只是感情好蹂风,可當我...
    茶點故事閱讀 67,974評論 6 393
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著乾蓬,像睡著了一般惠啄。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上任内,一...
    開封第一講書人閱讀 51,754評論 1 307
  • 那天撵渡,我揣著相機與錄音,去河邊找鬼族奢。 笑死姥闭,一個胖子當著我的面吹牛丹鸿,可吹牛的內容都是我干的越走。 我是一名探鬼主播,決...
    沈念sama閱讀 40,464評論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼廊敌!你這毒婦竟也來了铜跑?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 39,357評論 0 276
  • 序言:老撾萬榮一對情侶失蹤骡澈,失蹤者是張志新(化名)和其女友劉穎锅纺,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體肋殴,經...
    沈念sama閱讀 45,847評論 1 317
  • 正文 獨居荒郊野嶺守林人離奇死亡囤锉,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,995評論 3 338
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了护锤。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片官地。...
    茶點故事閱讀 40,137評論 1 351
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖烙懦,靈堂內的尸體忽然破棺而出驱入,到底是詐尸還是另有隱情,我是刑警寧澤氯析,帶...
    沈念sama閱讀 35,819評論 5 346
  • 正文 年R本政府宣布亏较,位于F島的核電站,受9級特大地震影響掩缓,放射性物質發(fā)生泄漏雪情。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,482評論 3 331
  • 文/蒙蒙 一你辣、第九天 我趴在偏房一處隱蔽的房頂上張望旺罢。 院中可真熱鬧,春花似錦绢记、人聲如沸扁达。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,023評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽跪解。三九已至,卻和暖如春签孔,著一層夾襖步出監(jiān)牢的瞬間叉讥,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,149評論 1 272
  • 我被黑心中介騙來泰國打工饥追, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留图仓,地道東北人。 一個月前我還...
    沈念sama閱讀 48,409評論 3 373
  • 正文 我出身青樓但绕,卻偏偏與公主長得像救崔,于是被迫代替她去往敵國和親惶看。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 45,086評論 2 355

推薦閱讀更多精彩內容