File size: 858 Bytes
3bbb319 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
# raw Makefile configuration
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)"
fi
if ! $WITH_IO ; then
LINE "USE_OPENCV := 0"
LINE "USE_LEVELDB := 0"
LINE "USE_LMDB := 0"
fi
if $WITH_CUDA ; then
# Only build SM50
LINE "CUDA_ARCH := -gencode arch=compute_50,code=sm_50"
else
LINE "CPU_ONLY := 1"
fi
if $WITH_CUDNN ; then
LINE "USE_CUDNN := 1"
fi
|