4dff33a559b0fc6837236c8003877bd915737e7c
[jacinto-ai/caffe-jacinto.git] / Makefile.config
1 ## Refer to http://caffe.berkeleyvision.org/installation.html
2 # Contributions simplifying and improving our build system are welcome!
4 # cuDNN acceleration switch (uncomment to build with cuDNN).
5 # cuDNN version 6 or higher is required.
6 USE_CUDNN := 1
8 # NCCL acceleration switch (uncomment to build with NCCL)
9 # See https://github.com/NVIDIA/nccl
10 USE_NCCL := 1
12 # CPU-only switch (uncomment to build without GPU support).
13 # Disables FP16 support.
14 # CPU_ONLY := 1
15 # Builds tests with 16 bit float support in addition to 32 and 64 bit.
16 # TEST_FP16 := 1
18 # uncomment to disable IO dependencies and corresponding data layers
19 # USE_OPENCV := 0
20 # USE_LEVELDB := 0
21 # USE_LMDB := 0
23 # Uncomment if you're using OpenCV 3
24 # OPENCV_VERSION := 3
26 # To customize your choice of compiler, uncomment and set the following.
27 # N.B. the default for Linux is g++ and the default for OSX is clang++
28 # CUSTOM_CXX := g++
30 # CUDA directory contains bin/ and lib/ directories that we need.
31 CUDA_DIR := /usr/local/cuda
32 # On Ubuntu 14.04, if cuda tools are installed via
33 # "sudo apt-get install nvidia-cuda-toolkit" then use this instead:
34 # CUDA_DIR := /usr
36 # CUDA architecture setting: going with all of them.
37 CUDA_ARCH :=    -gencode arch=compute_50,code=sm_50 \
38                 -gencode arch=compute_52,code=sm_52 \
39                 -gencode arch=compute_60,code=sm_60 \
40                 -gencode arch=compute_61,code=sm_61 \
41                 -gencode arch=compute_61,code=compute_61
43 # BLAS choice:
44 # atlas for ATLAS
45 # mkl for MKL
46 # open for OpenBlas - default, see https://github.com/xianyi/OpenBLAS
47 BLAS := open
48 # Custom (MKL/ATLAS/OpenBLAS) include and lib directories.
49 BLAS_INCLUDE := /opt/OpenBLAS/include/
50 BLAS_LIB := /opt/OpenBLAS/lib/
52 # Homebrew puts openblas in a directory that is not on the standard search path
53 # BLAS_INCLUDE := $(shell brew --prefix openblas)/include
54 # BLAS_LIB := $(shell brew --prefix openblas)/lib
56 # This is required only if you will compile the matlab interface.
57 # MATLAB directory should contain the mex binary in /bin.
58 # MATLAB_DIR := /usr/local
59 # MATLAB_DIR := /Applications/MATLAB_R2012b.app
61 # NOTE: this is required only if you will compile the python interface.
62 # We need to be able to find Python.h and numpy/arrayobject.h.
63 PYTHON_INCLUDE := /usr/include/python2.7 \
64                 /usr/lib/python2.7/dist-packages/numpy/core/include
65 # Anaconda Python distribution is quite popular. Include path:
66 # Verify anaconda location, sometimes it's in root.
67 # ANACONDA_HOME := $(HOME)/anaconda
68 # PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
69                 # $(ANACONDA_HOME)/include/python2.7 \
70                 # $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \
72 # Uncomment to use Python 3 (default is Python 2)
73 # PYTHON_LIBRARIES := boost_python3 python3.5m
74 # PYTHON_INCLUDE := /usr/include/python3.5m \
75 #                 /usr/lib/python3.5/dist-packages/numpy/core/include
77 # We need to be able to find libpythonX.X.so or .dylib.
78 PYTHON_LIB := /usr/lib
79 # PYTHON_LIB := $(ANACONDA_HOME)/lib
81 # Homebrew installs numpy in a non standard path (keg only)
82 # PYTHON_INCLUDE += $(dir $(shell python -c 'import numpy.core; print(numpy.core.__file__)'))/include
83 # PYTHON_LIB += $(shell brew --prefix numpy)/lib
85 # Uncomment to support layers written in Python (will link against Python libs)
86 WITH_PYTHON_LAYER := 1
88 # Whatever else you find you need goes here.
89 INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
90 LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial
92 # If Homebrew is installed at a non standard location (for example your home directory) and you use it for general dependencies
93 # INCLUDE_DIRS += $(shell brew --prefix)/include
94 # LIBRARY_DIRS += $(shell brew --prefix)/lib
96 # Uncomment to use `pkg-config` to specify OpenCV library paths.
97 # (Usually not necessary -- OpenCV libraries are normally installed in one of the above $LIBRARY_DIRS.)
98 # USE_PKG_CONFIG := 1
100 BUILD_DIR := build
101 DISTRIBUTE_DIR := distribute
103 # Uncomment for debugging. Does not work on OSX due to https://github.com/BVLC/caffe/issues/171
104 # DEBUG := 1
106 # The ID of the GPU that 'make runtest' will use to run unit tests.
107 TEST_GPUID := 0
109 # enable pretty build (comment to see full commands)
110 Q ?= @
112 # shared object suffix name to differentiate branches
113 LIBRARY_NAME_SUFFIX := -nv