Caffe安裝教程:Ubuntu16.04(CPU)
本文是在新安裝的Ubuntu16.04上進行Caffe的安裝,每個版本的Ubuntu安裝略有不同,請自行百度菠隆。
一、檢查自己電腦是否具有GPU
通常Caffe在計算時有兩種模式可以選擇狂秘,CPU或GPU骇径,使用GPU處理圖像速度會更快,但往往有的計算機沒有GPU者春,配置太低破衔,所以只能選擇CPU,作者的電腦不支持GPU钱烟,因此選擇CPU安裝的版本晰筛。
輸入:
lspci | grep -i nvidia
<dl>
<dt>結(jié)果1</dt>
<dd>如果未顯示任何內(nèi)容,則證明你的電腦不支持GPU拴袭。</dd>
<dt>結(jié)果2</dt>
<dd>如果顯示如下圖內(nèi)容:
則表明版本是GTX 650读第,支持GPU,可以去http://developer.nvidia.com/cuda-gpus去驗證稻扬,支持CUDA,本文只描述CPU模式的caffe安裝羊瘩,如果要安裝GPU模式的caffe泰佳,請自行百度或Google盼砍。</dd>
</dl>
二、依次安裝依賴包
依次執(zhí)行以下語句逝她,安裝依賴包:
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 libhdf5-serial-dev
sudo apt-get install protobuf-compiler
sudo apt-get install libgflags-dev
sudo apt-get install libgoogle-glog-dev
sudo apt-get install liblmdb-dev
sudo apt-get install libatlas-base-dev
三浇坐、下載caffe
可直接使用Git下載caffe,如果沒有Git黔宛,請先安裝Git近刘,作者新裝的Ubuntu16.04沒有Git,所以先安裝Git臀晃,執(zhí)行如下語句:
sudo apt-get install git
安裝結(jié)束后下載caffe觉渴,執(zhí)行以下命令:
git clone git://github.com/BVLC/caffe.git
這樣你就可以在根目錄下看到有個caffe目錄了
四、編譯caffe
1.進入caffe目錄
cd caffe/
2.生成Makefile.config文件徽惋,這里是將caffe目錄下自帶的Makefile.config.example文件復制一份并更名為Makefile.config案淋,命令如下:
cp Makefile.config.example Makefile.config
此時目錄下就會有Makefile.config文件,如下圖:
3.修改Makefile.config文件中的配置
1)編輯Makefile.config文件
sudo gedit Makefile.config
2)去掉CPU_ONLY前面的#號
使得CPU_ONLY := 1
3)配置引用文件路徑(主要是HDF5的路徑問題)
原來的路徑配置如下圖:
[圖片上傳失敗...(image-b60a7f-1512963090846)]
修改為如下圖:
新增的內(nèi)容為:
/usr/include/hdf5/serial
/usr/lib/x86_64-linux-gnu/hdf5/serial
4)執(zhí)行編譯
sudo make all
sudo make test
sudo make runtest
執(zhí)行玩runtest后险绘,會有結(jié)果顯示:
顯示1098個用例執(zhí)行成功踢京,結(jié)果PASSED!
注意:如果執(zhí)行結(jié)果失敗宦棺,則需要執(zhí)行語句sudo make clean
,然后解決掉問題重新編譯
五瓣距、編譯python接口
通常為了更好的調(diào)用caffe,我們往往需要寫代碼代咸,caffe具有Python和C++接口蹈丸,我們較為常用Python接口,這里我將展示如何編譯Python接口侣背。
1)安裝pip
因為Ubuntu16.04默認安裝的是python2.7白华,沒有pip,所以需要先安裝pip贩耐,命令如下:
sudo apt-get install python-pip
2)安裝Python接口依賴庫
在caffe根目錄下弧腥,有個python文件夾,文件夾里面有個requirements.txt潮太,里面有需要的依賴庫和版本信息管搪,按照其安裝即可,在安裝前铡买,需要先安裝fortran編輯器(gfrotran)更鲁,因為安裝scipy庫時需要它,命令如下:
sudo apt-get install gfortran
cd ~/caffe/python
for req in $(cat requirements.txt); do pip install $req; done
安裝結(jié)束后奇钞,可以執(zhí)行如下語句驗證:
sudo pip install -r requirements.txt
可以看到澡为,安裝成功的,都會顯示Requirement already satisfied景埃,沒安裝成功的媒至,會繼續(xù)進行安裝顶别。
3)將caffe根目錄下的python文件夾加入到環(huán)境變量
先打開配置文件bashrc
sudo gedit ~/.bashrc
在文件的最后面添加
export PYTHONPATH=/home/moqi/caffe/python:$PYTHONPATH
如下圖:
注意,這里路徑根據(jù)每個人電腦則不一樣拒啰,例如驯绎,我的下載的caffe在/home/moqi/下面,所以路徑為/home/moqi/caffe/python谋旦,你可以自行進入自己的caffe文件夾里的python目錄下剩失,輸入pwd查看路徑。
然后執(zhí)行更新配置(有時更新不起作用時册着,可以重啟電腦就會有效):
sudo ldconfig
4)編譯python接口
cd ~/caffe/
make pycaffe
5)驗證python接口
進行python環(huán)境拴孤,引入caffe包,如果沒有報錯則安裝成功指蚜!
如下圖:
至此乞巧,caffe安裝成功!
六摊鸡、題外話
因為系統(tǒng)或者安裝機器的不同绽媒,可能會出現(xiàn)一些不一樣的問題,出現(xiàn)問題免猾,請粘貼問題自行g(shù)oogle或百度是辕,都是有答案的,切忌眼高手低猎提。