服務器(多GPU)caffe安裝和編譯

refer to

一. 前提

多GPU交互在神經網絡是常見的薄辅,所以在安裝caffe之前需要安裝NCCL,來保證多GPU之間的相互交流摄闸。
多GPU禁漓,這里指的是2個及2個以上英偉達顯卡,而不是筆記本中的集顯和獨顯忧侧。

二.安裝NCCL

**1.下載編譯 **
shell終端

cd nccl
make CUDA_HOME=/user/local/cuda-7.5   test #注意自己的cuda路徑

2.測試和配置環(huán)境變量
shell終端

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:./build/lib
./build/test/single/all_reduce_test
./build/test/single/all_reduce_test 10000000
make install

注:make install 是自己添加石窑,而官方原文沒有。之所以這么加是因為在caffe 執(zhí)行 cmake時候蚓炬,cmake無法找到
非deb安裝軟件的路徑松逊,所以添加make install 是為了能讓cmake識別到路徑。

三.安裝caffe

安裝依賴&下載caffe (略)
編譯

cp Makefile.config.example Makefile.config
gedit Makefile.config

打開文本后肯夏,作出如下修改
取消下面這些話的前面注釋符號#

USE_CUDNN := 1 
USE_NCCL := 1
ANACONDA_HOME := $(HOME)/anaconda   #這里我們使用Anaconda環(huán)境下的python
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
         $(ANACONDA_HOME)/include/python2.7 \
         $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \
PYTHON_LIB := $(ANACONDA_HOME)/lib

在下面這些語句中加上#注釋符號

#PYTHON_INCLUDE := /usr/include/python2.7 \
        # /usr/lib/python2.7/dist-packages/numpy/core/include
#PYTHON_LIB := /usr/lib

保存经宏,退出,編譯caffe

sudo pip install -r caffe/python/requirements.txt
cd caffe
mkdir build
cd build
make all -j
make install -j
make runtest -j
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
  • 序言:七十年代末驯击,一起剝皮案震驚了整個濱河市烁兰,隨后出現的幾起案子,更是在濱河造成了極大的恐慌徊都,老刑警劉巖沪斟,帶你破解...
    沈念sama閱讀 212,599評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現場離奇詭異碟贾,居然都是意外死亡币喧,警方通過查閱死者的電腦和手機,發(fā)現死者居然都...
    沈念sama閱讀 90,629評論 3 385
  • 文/潘曉璐 我一進店門袱耽,熙熙樓的掌柜王于貴愁眉苦臉地迎上來杀餐,“玉大人,你說我怎么就攤上這事朱巨∈非蹋” “怎么了?”我有些...
    開封第一講書人閱讀 158,084評論 0 348
  • 文/不壞的土叔 我叫張陵冀续,是天一觀的道長琼讽。 經常有香客問我,道長洪唐,這世上最難降的妖魔是什么钻蹬? 我笑而不...
    開封第一講書人閱讀 56,708評論 1 284
  • 正文 為了忘掉前任,我火速辦了婚禮凭需,結果婚禮上问欠,老公的妹妹穿的比我還像新娘肝匆。我一直安慰自己,他們只是感情好顺献,可當我...
    茶點故事閱讀 65,813評論 6 386
  • 文/花漫 我一把揭開白布旗国。 她就那樣靜靜地躺著,像睡著了一般注整。 火紅的嫁衣襯著肌膚如雪能曾。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 50,021評論 1 291
  • 那天肿轨,我揣著相機與錄音寿冕,去河邊找鬼。 笑死萝招,一個胖子當著我的面吹牛蚂斤,可吹牛的內容都是我干的。 我是一名探鬼主播槐沼,決...
    沈念sama閱讀 39,120評論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼曙蒸,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了岗钩?” 一聲冷哼從身側響起纽窟,我...
    開封第一講書人閱讀 37,866評論 0 268
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎兼吓,沒想到半個月后臂港,有當地人在樹林里發(fā)現了一具尸體,經...
    沈念sama閱讀 44,308評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡视搏,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 36,633評論 2 327
  • 正文 我和宋清朗相戀三年审孽,在試婚紗的時候發(fā)現自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片浑娜。...
    茶點故事閱讀 38,768評論 1 341
  • 序言:一個原本活蹦亂跳的男人離奇死亡佑力,死狀恐怖,靈堂內的尸體忽然破棺而出筋遭,到底是詐尸還是另有隱情打颤,我是刑警寧澤,帶...
    沈念sama閱讀 34,461評論 4 333
  • 正文 年R本政府宣布漓滔,位于F島的核電站编饺,受9級特大地震影響,放射性物質發(fā)生泄漏响驴。R本人自食惡果不足惜透且,卻給世界環(huán)境...
    茶點故事閱讀 40,094評論 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望豁鲤。 院中可真熱鬧石蔗,春花似錦罕邀、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,850評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽日熬。三九已至棍厌,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間竖席,已是汗流浹背耘纱。 一陣腳步聲響...
    開封第一講書人閱讀 32,082評論 1 267
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留毕荐,地道東北人束析。 一個月前我還...
    沈念sama閱讀 46,571評論 2 362
  • 正文 我出身青樓,卻偏偏與公主長得像憎亚,于是被迫代替她去往敵國和親员寇。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 43,666評論 2 350

推薦閱讀更多精彩內容