Makefile.config.example

   1 ## Refer to http://caffe.berkeleyvision.org/installation.html
   2 # Contributions simplifying and improving our build system are welcome!
   3
   4 # cuDNN acceleration switch (comment to build without cuDNN).
   5 USE_CUDNN := 1
   6
   7 # CPU-only switch (uncomment to build without GPU support).
   8 # CPU_ONLY := 1
   9
  10 # To customize your choice of compiler, uncomment and set the following.
  11 # N.B. the default for Linux is g++ and the default for OSX is clang++
  12 # CUSTOM_CXX := g++
  13
  14 # CUDA directory contains bin/ and lib/ directories that we need.
  15 CUDA_DIR := /usr/local/cuda
  16 # On Ubuntu 14.04, if cuda tools are installed via
  17 # "sudo apt-get install nvidia-cuda-toolkit" then use this instead:
  18 # CUDA_DIR := /usr
  19
  20 # CUDA architecture setting: going with all of them.
  21 # For CUDA < 6.0, comment the *_50 lines for compatibility.
  22 CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
  23                 -gencode arch=compute_20,code=sm_21 \
  24                 -gencode arch=compute_30,code=sm_30 \
  25                 -gencode arch=compute_35,code=sm_35 \
  26                 -gencode arch=compute_50,code=sm_50 \
  27                 -gencode arch=compute_50,code=compute_50
  28
  29 # BLAS choice:
  30 # atlas for ATLAS (default)
  31 # mkl for MKL
  32 # open for OpenBlas
  33 BLAS := atlas
  34 # Custom (MKL/ATLAS/OpenBLAS) include and lib directories.
  35 # Leave commented to accept the defaults for your choice of BLAS
  36 # (which should work)!
  37 # BLAS_INCLUDE := /path/to/your/blas
  38 # BLAS_LIB := /path/to/your/blas
  39
  40 # This is required only if you will compile the matlab interface.
  41 # MATLAB directory should contain the mex binary in /bin.
  42 # MATLAB_DIR := /usr/local
  43 # MATLAB_DIR := /Applications/MATLAB_R2012b.app
  44
  45 # NOTE: this is required only if you will compile the python interface.
  46 # We need to be able to find Python.h and numpy/arrayobject.h.
  47 PYTHON_INCLUDE := /usr/include/python2.7 \
  48                 /usr/lib/python2.7/dist-packages/numpy/core/include \
  49                 /usr/local/lib/python2.7/dist-packages/numpy/core/include
  50 # Anaconda Python distribution is quite popular. Include path:
  51 # Verify anaconda location, sometimes it's in root.
  52 # ANACONDA_HOME := $(HOME)/anaconda
  53 # PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
  54                 # $(ANACONDA_HOME)/include/python2.7 \
  55                 # $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \
  56
  57 # We need to be able to find libpythonX.X.so or .dylib.
  58 PYTHON_LIB := /usr/lib
  59 # PYTHON_LIB := $(ANACONDA_HOME)/lib
  60
  61 # Uncomment to support layers written in Python (will link against Python libs)
  62 # WITH_PYTHON_LAYER := 1
  63
  64 # Whatever else you find you need goes here.
  65 INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
  66 LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
  67
  68 # Uncomment to use `pkg-config` to specify OpenCV library paths.
  69 # (Usually not necessary -- OpenCV libraries are normally installed in one of the above $LIBRARY_DIRS.)
  70 # USE_PKG_CONFIG := 1
  71
  72 BUILD_DIR := build
  73 DISTRIBUTE_DIR := distribute
  74
  75 # Uncomment for debugging. Does not work on OSX due to https://github.com/BVLC/caffe/issues/171
  76 # DEBUG := 1
  77
  78 # The ID of the GPU that 'make runtest' will use to run unit tests.
  79 TEST_GPUID := 0
  80
  81 # enable pretty build (comment to see full commands)
  82 Q ?= @
  83
  84 # Adds timing info in logs
  85 # BENCHMARK_DATA := 1
  86 # BENCHMARK_SOLVER := 1
  87
  88 # shared object suffix name to differentiate branches
  89 LIBRARY_NAME_SUFFIX := -nv