您的位置：首页 > Web前端

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

2017-08-26 20:07 645 查看
编译cpu版caffe

从caffe项目主页把caffe项目clone下来

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


然后：

cd caffe
cp 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/serial
LIBRARY_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 all
make test
make runtest
make pycaffe


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

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


若没有错误则表示安装成功，否则make clean，重新编译。
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息
标签：  ubuntu python cpu ssd opencv
相关文章推荐
新的分享
章节导航