Ubuntu14.04安装CPU版SSD(Single Shot MultiBox Detector)/Caffe版本(二)

来源:互联网 发布:迈网络摄像头默认ip 编辑:程序博客网 时间:2024/06/14 00:57

编译cpu版caffe
从caffe项目主页把caffe项目clone下来

git clone --recursive https://github.com/BVLC/caffe.git 

然后:

cd caffecp Makefile.config.example Makefile.config 

由于是仅CPU安装,修改Makefile相关配置:
去掉注释CPU_ONLY :=1
注释掉CUDA有关的行:

#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 

去掉注释WITH_PYTHON_LAYER := 1

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serialLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/i386-linux-gnu/hdf5/serial /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial#TEST_GPUID := 0 

文件修改完成后,开始编译:

make allmake testmake runtestmake pycaffe 

若编译没有错误,则编译成功。完成后:

$cd caffe/python$python>>>import caffe 

若没有错误则表示安装成功,否则make clean,重新编译。

阅读全文
0 0
原创粉丝点击