| | |
| |
|
| | 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 |
| |
|
| |
|