新聞中心
安裝環(huán)境:ubuntu 16.04 lts 64位, gcc5.4 gpu1080ti,cuda8.0,cudnn5.1.10

1. 安裝依賴庫
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev libboost-all-dev protobuf-compiler sudo apt-get install libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-dev
2. 安裝caffe
終端輸入命令:
git clone git://github.com/BVLC/caffe.git
然后找到caffe文件夾,打開后會發(fā)現(xiàn)Makefile.config.example 文件,
這時候?qū)akefile.config.example復制一份命名為Makefile.config,并打開,將
#USE_CUDNN := 1 # CUDA_DIR := /usr/local/cuda # CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \ #-gencode arch=compute_20,code=sm_21 \ #-gencode arch=compute_30,code=sm_30 \ #-gencode arch=compute_35,code=sm_35 \ #-gencode arch=compute_50,code=sm_50 \ #-gencode arch=compute_50,code=compute_50
去掉注釋#,并且CUDA_DIR 更換為你的cuda安裝路徑
打開Makefile.config文件,將
/usr/include/hdf5/serial
添加到
INCLUDE_DIRS := $(Python_INCLUDE)/usr/local/include
打開Makefile文件,找到
LIBRARIES +=glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
更改為:
LIBRARIES +=glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
之后執(zhí)行
make all make test make runtest
若在編譯過程中報錯出現(xiàn)丟失文件的情況,可以尋找丟失文件路徑。
例如常見問題:
./include/caffe/util/hdf5.hpp:6:18: fatal error: hdf5.h: No such file or directory
解決方法:找到文件路徑:
sudo find / -name hdf5.h
找到對應(yīng)文件路徑,將其加入Makefile.config中。
查詢結(jié)果為:
/usr/include/hdf5/serial/hdf5.h
打開Makefile.config,在
INCLUDE_DIRS := $(PYTHON_INCLUDE)/usr/local/include
之后添加
/usr/include/hdf5/serial
注意路徑,有INCLUDE_DIRS 和 LIBRARY_DIRS兩個,看清丟失文件的路徑在/usr/include中還是/usr/lib中
正常情況下應(yīng)該是沒問題了。
3. 安裝anaconda,配置python caffe接口
1.安裝anaconda ,選擇Python2.7 注意文件名
bash Anaconda2-4.3.0-Linux-x86_64.sh
2.安裝過程中注意提示,千萬不要一直enter,記得一直選擇默認的就行,傻瓜式安裝,之后會提示是否在bashrc文檔中添加路徑,這里選擇“yes”
3.打開bashrc文檔,在Home下,打開顯示隱藏文件選項,就可以找到該文件。在最后一行里加入
export PATH=/home/(你的用戶名)/anaconda2/bin:$PATH(具體路徑根據(jù)你的安裝路徑而定) export PYTHONPATH=/home/(你的用戶名)/caffe/python:$PYTHONPATH(同上) export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libstdc++.so.6
Bashrc文檔更改后需要source一下,或者注銷/重啟電腦:
4. .修改caffe下的Makefile.config文檔
#ANACONDA_HOME := $(HOME)/anaconda2 #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 #WITH_PYTHON_LAYER := 1
去掉注釋#,將
PYTHON_LIB := /usr/lib
加上注釋。
5. 升級Anaconda中的gcc版本
conda update --all
或者
conda install libgcc
出現(xiàn)升級確認提示,輸入y,升級成功后,關(guān)閉終端重新打開
6. 回到caffe目錄下,編譯
make all make pycaffe
成功后,輸入python,接著輸入
import caffe
出現(xiàn)錯誤提示:
No module named google.protobuf
此時輸入命令
conda install protobuf
結(jié)束后再重新import caffe,成功的話應(yīng)該沒有任何提示。
4. 測試
進入caffe安裝目錄
./data/mnist/get_mnist.sh ./examples/mnist/create_mnist.sh ./examples/mnist/train_lenet.sh
成功的話那么久就可以結(jié)束安裝了!
文章題目:Ubuntu16.04安裝CaffeGPU版
網(wǎng)址分享:http://www.dlmjj.cn/article/ccscesc.html


咨詢
建站咨詢
