|
|
|
|
|
LINE () { |
|
echo "$@" >> Makefile.config |
|
} |
|
|
|
cp Makefile.config.example Makefile.config |
|
|
|
LINE "BLAS := open" |
|
LINE "WITH_PYTHON_LAYER := 1" |
|
|
|
if $WITH_PYTHON3 ; then |
|
|
|
|
|
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 |
|
|
|
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 |
|
|
|
|