1.下载caffe-ssd
cd ssd # ssd是一个空的文件
git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd #检查分支
2.编译前准备:Makefile.config
cp Makefile.config.example Makefile.config
不谈版本的编译都不是好编译!!
Python3.5+cudnn6.0.21+cuda8.0.61+opencv3.30
打开Makefile.config:
USE_CUDNN := 1
OPENCV_VERSION := 3
CUDA_DIR := /usr/local/cuda
CUDA_ARCH := -gencode arch=compute_30,code=sm_30 \
-gencode arch=compute_35,code=sm_35 \
-gencode arch=compute_50,code=sm_50 \
-gencode arch=compute_52,code=sm_52 \
-gencode arch=compute_60,code=sm_60 \
-gencode arch=compute_61,code=sm_61 \
-gencode arch=compute_61,code=compute_61
BLAS := atlas
PYTHON_LIBRARIES := boost_python-py35 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5m \ /usr/lib/python3.5/dist-packages/numpy/core/include \ /usr/local/lib/python3.5/dist-packages/numpy/core/include
PYTHON_LIB := /usr/lib
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/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
USE_NCCL := 1
BUILD_DIR := build
DISTRIBUTE_DIR := distribute
TEST_GPUID := 0 Q ?= @
2.编译:
make all -j8
make runtest -j8
make pycaffe -j8