LINE () {

echo "$@" >> Makefile.config

}

cp Makefile.config.example Makefile.config

LINE "BLAS := open"

LINE "WITH_PYTHON_LAYER := 1"

if $WITH_PYTHON3 ; then

# TODO(lukeyeager) this path is currently disabled because of test errors like:

# ImportError: dynamic module does not define init function (PyInit__caffe)

LINE "PYTHON_LIBRARIES := python3.4m boost_python-py34"

LINE "PYTHON_INCLUDE := /usr/include/python3.4 /usr/lib/python3/dist-packages/numpy/core/include"

LINE "INCLUDE_DIRS := \$(INCLUDE_DIRS) \$(PYTHON_INCLUDE)"

if ! $WITH_IO ; then

LINE "USE_OPENCV := 0"

LINE "USE_LEVELDB := 0"

LINE "USE_LMDB := 0"

if $WITH_CUDA ; then

# Only build SM50

LINE "CUDA_ARCH := -gencode arch=compute_50,code=sm_50"

LINE "CPU_ONLY := 1"

if $WITH_CUDNN ; then

LINE "USE_CUDNN := 1"