SSD的训练 Linux下

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
上一篇:3D局部光照模型


下一篇:JetsonNano安装Caffe