Ubuntu16.04+CUDA8.0+Caffe

[TOC]

簡介

caffe是一個清晰产阱,可讀性高块仆,快速的深度學習框架构蹬。作者是賈揚清榨乎,加州大學伯克利的ph.D,現(xiàn)就職于Facebook蜜暑。caffe的官網是http://caffe.berkeleyvision.org/

安裝

環(huán)境

  • Ubuntu 16.04 LTS
  • CUDA8.0

參考資料

注: 對于 Ubuntu >= 17.04 的Ubuntu系統(tǒng)肛捍,安裝將變得極為簡單,通過Ubuntu的包管理器安裝即可拙毫,這里不作說明。

準備

基礎環(huán)境配置

依賴庫安裝

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

構建與安裝

獲取源碼

這里 (或終端輸入命令 git clone https://github.com/BVLC/caffe.git )下載caffe源碼并解壓蛀醉,修改文件夾名為 caffe

For Python

如果你需要使用Caffe 的 Python 接口拯刁,你還需要安裝Python的一些庫,進入caffe根目錄下的python目錄垛玻,執(zhí)行:

for req in $(cat requirements.txt); do sudo pip install $req; done

然后添加 PYTHONPATH 環(huán)境變量:終端輸入 sudo gedit ~/.bashrc 打開 .bashrc 文件奶躯,文件末尾加入(注意修改為你的caffe絕對路徑):

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

如果你使用 Anaconda 帚桩,下面修改 Makefile.config 文件時巫糙,注意自行 啟用 Anaconda,約73行。

修改配置文件

這一步修改 Makefile.config 文件醉锄,以滿足自己的需求,如:

#啟用cuDNN加速
USE_CUDNN := 1

另外檩小,在Ubuntu16.04上安裝caffe還需要修改以下幾處,否則會報出各種錯誤规求,具體參見問題解決

將如下幾處代碼:

#第一處:CUDA PATH
CUDA_DIR := /usr/local/cuda

#第二處:PYTHON PATH
# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include

改為:

#第一處:CUDA PATH卵惦,改為:
CUDA_DIR := /usr/local/cuda-8.0

#第二處:PYTHON PATH阻肿,添加hdf5庫沮尿,改為:
# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/

構建安裝

進入caffe的根目錄,終端執(zhí)行如下命令畜疾,不報錯誤,代表安裝成功啡捶!( -j$(nproc) 代表使用最大的線程編譯,當然也可以手動指定彤敛,如 make -j4make py 是用來編譯caffe的python接口了赌,如不需要可忽略。)

make all -j$(nproc)
make pycaffe -j$(nproc)
make test -j$(nproc)
make runtest -j$(nproc)

添加環(huán)境變量

終端輸入 sudo gedit ~/.bashrc 打開 ".bashrc" 文件揍拆,在文件末尾加入如下代碼并保存:

# Set Caffe environment
export CAFFE_ROOT=/home/liu/sfw/dlapp/caffe/

#caffe python
export PYTHONPATH=/home/liu/sfw/dlapp/caffe/python/:$PYTHONPATH

然后重新打開一個終端茶凳,或者輸入 source ~/.bashrc 加載新的環(huán)境變量。

驗證Python接口

為驗證安裝成功贮喧,終端輸入 ipythonpython 進入Python解釋器環(huán)境筒狠,輸入: import caffe 箱沦,不報錯誤代表caffe的Python接口可以使用辩恼。

問題解決

安裝

hdf5

  1. 現(xiàn)象
In file included from src/caffe/solvers/sgd_solver.cpp:5:0:
./include/caffe/util/hdf5.hpp:6:18: fatal error: hdf5.h: No such file or directory
compilation terminated.
Makefile:581: recipe for target '.build_release/src/caffe/solvers/sgd_solver.o' failed
  1. 解決

首先確保前面安裝依賴庫時安裝了 libhdf5-serial-dev ,然后修改 Makefile.config 文件(約95行):

# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/

No module named caffe

請確認是否執(zhí)行 make pycaffe 且沒有報錯灶伊,如果是請再檢查環(huán)境變量的設置。

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末竹椒,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子胸完,更是在濱河造成了極大的恐慌,老刑警劉巖翘贮,帶你破解...
    沈念sama閱讀 206,311評論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異锨能,居然都是意外死亡肴捉,警方通過查閱死者的電腦和手機腹侣,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,339評論 2 382
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來傲隶,“玉大人窃页,你說我怎么就攤上這事跺株〔甭簦” “怎么了?”我有些...
    開封第一講書人閱讀 152,671評論 0 342
  • 文/不壞的土叔 我叫張陵畦木,是天一觀的道長。 經常有香客問我十籍,道長,這世上最難降的妖魔是什么勾栗? 我笑而不...
    開封第一講書人閱讀 55,252評論 1 279
  • 正文 為了忘掉前任,我火速辦了婚禮砸讳,結果婚禮上,老公的妹妹穿的比我還像新娘簿寂。我一直安慰自己,他們只是感情好陶耍,可當我...
    茶點故事閱讀 64,253評論 5 371
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著烈钞,像睡著了一般。 火紅的嫁衣襯著肌膚如雪馒过。 梳的紋絲不亂的頭發(fā)上酗钞,一...
    開封第一講書人閱讀 49,031評論 1 285
  • 那天腹忽,我揣著相機與錄音砚作,去河邊找鬼。 笑死着裹,一個胖子當著我的面吹牛,可吹牛的內容都是我干的骇扇。 我是一名探鬼主播,決...
    沈念sama閱讀 38,340評論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼少孝,長吁一口氣:“原來是場噩夢啊……” “哼熬苍!你這毒婦竟也來了?” 一聲冷哼從身側響起柴底,我...
    開封第一講書人閱讀 36,973評論 0 259
  • 序言:老撾萬榮一對情侶失蹤似枕,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后年柠,有當地人在樹林里發(fā)現(xiàn)了一具尸體褪迟,經...
    沈念sama閱讀 43,466評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡味赃,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 35,937評論 2 323
  • 正文 我和宋清朗相戀三年虐拓,在試婚紗的時候發(fā)現(xiàn)自己被綠了蓉驹。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,039評論 1 333
  • 序言:一個原本活蹦亂跳的男人離奇死亡狠持,死狀恐怖瞻润,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情正勒,我是刑警寧澤傻铣,帶...
    沈念sama閱讀 33,701評論 4 323
  • 正文 年R本政府宣布,位于F島的核電站阱驾,受9級特大地震影響怪蔑,放射性物質發(fā)生泄漏。R本人自食惡果不足惜喧枷,卻給世界環(huán)境...
    茶點故事閱讀 39,254評論 3 307
  • 文/蒙蒙 一隧甚、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧戚扳,春花似錦族吻、人聲如沸珠增。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,259評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至梦皮,卻和暖如春孤澎,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背退子。 一陣腳步聲響...
    開封第一講書人閱讀 31,485評論 1 262
  • 我被黑心中介騙來泰國打工型将, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人丸凭。 一個月前我還...
    沈念sama閱讀 45,497評論 2 354
  • 正文 我出身青樓腕铸,卻偏偏與公主長得像狠裹,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子涛菠,可洞房花燭夜當晚...
    茶點故事閱讀 42,786評論 2 345

推薦閱讀更多精彩內容