ubuntu16.04 安装caffe以及python接口
http://blog.csdn.net/qq_25073253/article/details/72571714
http://blog.csdn.net/greed7480/article/details/68486039
首先在你要安裝的路徑下 clone :
git clone https://github.com/BVLC/caffe.git
進(jìn)入 caffe ,將 Makefile.config.example 文件復(fù)制一份并更名為 Makefile.config ,也可以在 caffe 目錄下直接調(diào)用以下命令完成復(fù)制操作 :
sudo cp Makefile.config.example Makefile.config
復(fù)制一份的原因是編譯 caffe 時(shí)需要的是 Makefile.config 文件,而Makefile.config.example 只是caffe 給出的配置文件例子,不能用來(lái)編譯 caffe。
然后修改 Makefile.config 文件,在 caffe 目錄下打開(kāi)該文件:
sudo gedit Makefile.config
修改 Makefile.config 文件內(nèi)容:
1.應(yīng)用 cudnn
將
#USE_CUDNN := 1
修改成:
USE_CUDNN := 1
2.應(yīng)用 opencv 版本
將
#OPENCV_VERSION := 3
修改為:
OPENCV_VERSION := 3
3.使用 Python 接口
將
#WITH_PYTHON_LAYER := 1
修改為
WITH_PYTHON_LAYER := 1
4.修改 python 路徑
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
5 OK ,可以開(kāi)始編譯了,在 caffe 目錄下執(zhí)行 :
make all -j8
這是如果之前的配置或安裝出錯(cuò),那么編譯就會(huì)出現(xiàn)各種各樣的問(wèn)題,所以前面的步驟一定要細(xì)心。
編譯成功后可運(yùn)行測(cè)試:
make runtest -j8
其中,我最初在make runtest 的時(shí)候是用的指令$ sudo make runtest,結(jié)果就報(bào)
錯(cuò):
.build_release/tools/caffe: error while loading shared libraries: libcudart.so.8.0:
cannot open shared object file: No such file or directory
而換為$ make runtest 后就沒(méi)報(bào)錯(cuò)了,剛開(kāi)始特別納悶,為什么加了超級(jí)用戶(hù)權(quán)
限后反而還會(huì)有錯(cuò)誤呢,谷歌后才知道錯(cuò)誤里的“l(fā)ibcudart.so.8.0”是位于
“/usr/local/cuda-8.0/lib64”文件夾下的,而這個(gè)路徑我們之前已將其添加進(jìn)了
“LD_LIBRARY_PATH”環(huán)境變量中,但通過(guò)sudo 來(lái)執(zhí)行該命令可能會(huì)導(dǎo)致其無(wú)法
使用“LD_LIBRARY_PATH”環(huán)境變量,這只是我目前個(gè)人的猜測(cè),如果有清楚具
體原因的人,也請(qǐng)麻煩告知我一下,謝了!
6 添加python接口
首先編譯 pycaffe :
cd caffe
sudo make pycaffe -j8
7 環(huán)境變量
方法1 當(dāng)前用戶(hù)添加
sudo echo export PYTHONPATH="home/imagealg/software/caffe/python" >> ~/.bashrc
source ~/.bashrc
方法2 全局用戶(hù)添加
添加~/caffe/python到$PYTHONPATH:
$ sudo gedit /etc/profile
# 添加: export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH
$ source /etc/profile # 使之生效
?
8 測(cè)試
cd $CAFFE_ROOT
./data/mnist/get_mnist.sh
數(shù)據(jù)轉(zhuǎn)化
./examples/mnist/create_mnist.sh
訓(xùn)練模型
./examples/mnist/train_lenet.sh
測(cè)試模型
./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_10000.caffemodel -gpu=0
?
轉(zhuǎn)載于:https://www.cnblogs.com/adong7639/p/7843264.html
總結(jié)
以上是生活随笔為你收集整理的ubuntu16.04 安装caffe以及python接口的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: markdown一边写一边预览_mark
- 下一篇: DIGITS使用步骤