|
| 1 | +# This list is required for static linking and exported to CaffeConfig.cmake |
| 2 | +set(Caffe_LINKER_LIBS "") |
| 3 | + |
| 4 | +# ---[ Boost |
| 5 | +find_package(Boost 1.46 REQUIRED COMPONENTS system thread filesystem) |
| 6 | +include_directories(SYSTEM ${Boost_INCLUDE_DIR}) |
| 7 | +add_definitions(-DBOOST_ALL_NO_LIB) |
| 8 | +list(APPEND Caffe_LINKER_LIBS ${Boost_LIBRARIES}) |
| 9 | + |
| 10 | +if(DEFINED MSVC) |
| 11 | + # We should define this only when necessary, |
| 12 | + # i.e VS 2013 Update 4 or earlier. |
| 13 | + add_definitions(-DBOOST_NO_CXX11_TEMPLATE_ALIASES) |
| 14 | +endif() |
| 15 | + |
| 16 | +# ---[ Threads |
| 17 | +find_package(Threads REQUIRED) |
| 18 | +list(APPEND Caffe_LINKER_LIBS ${CMAKE_THREAD_LIBS_INIT}) |
| 19 | + |
| 20 | +# ---[ Google-glog |
| 21 | +include("cmake/External/glog.cmake") |
| 22 | +include_directories(SYSTEM ${GLOG_INCLUDE_DIRS}) |
| 23 | +list(APPEND Caffe_LINKER_LIBS ${GLOG_LIBRARIES}) |
| 24 | + |
| 25 | +# ---[ Google-gflags |
| 26 | +include("cmake/External/gflags.cmake") |
| 27 | +include_directories(SYSTEM ${GFLAGS_INCLUDE_DIRS}) |
| 28 | +list(APPEND Caffe_LINKER_LIBS ${GFLAGS_LIBRARIES}) |
| 29 | + |
| 30 | +# ---[ Google-protobuf |
| 31 | +include(cmake/ProtoBuf.cmake) |
| 32 | + |
| 33 | +# ---[ HDF5 |
| 34 | +if(MSVC) |
| 35 | + # Find HDF5 using it's hdf5-config.cmake file with MSVC |
| 36 | + if(DEFINED HDF5_DIR) |
| 37 | + list(APPEND CMAKE_MODULE_PATH ${HDF5_DIR}) |
| 38 | + endif() |
| 39 | + find_package(HDF5 COMPONENTS C HL REQUIRED) |
| 40 | +# set(HDF5_LIBRARIES hdf5-shared) |
| 41 | +# set(HDF5_HL_LIBRARIES hdf5_hl-shared) |
| 42 | +else() |
| 43 | + find_package(HDF5 COMPONENTS HL REQUIRED) |
| 44 | +endif() |
| 45 | +include_directories(SYSTEM ${HDF5_INCLUDE_DIRS} ${HDF5_HL_INCLUDE_DIR}) |
| 46 | +list(APPEND Caffe_LINKER_LIBS ${HDF5_LIBRARIES} ${HDF5_HL_LIBRARIES}) |
| 47 | + |
| 48 | +# ---[ LMDB |
| 49 | +if(USE_LMDB) |
| 50 | + find_package(LMDB REQUIRED) |
| 51 | + include_directories(SYSTEM ${LMDB_INCLUDE_DIR}) |
| 52 | + list(APPEND Caffe_LINKER_LIBS ${LMDB_LIBRARIES}) |
| 53 | + add_definitions(-DUSE_LMDB) |
| 54 | + if(ALLOW_LMDB_NOLOCK) |
| 55 | + add_definitions(-DALLOW_LMDB_NOLOCK) |
| 56 | + endif() |
| 57 | +endif() |
| 58 | + |
| 59 | +# ---[ LevelDB |
| 60 | +if(USE_LEVELDB) |
| 61 | + find_package(LevelDB REQUIRED) |
| 62 | + include_directories(SYSTEM ${LevelDB_INCLUDE}) |
| 63 | + list(APPEND Caffe_LINKER_LIBS ${LevelDB_LIBRARIES}) |
| 64 | + add_definitions(-DUSE_LEVELDB) |
| 65 | +endif() |
| 66 | + |
| 67 | +# ---[ Snappy |
| 68 | +if(USE_LEVELDB) |
| 69 | + find_package(Snappy REQUIRED) |
| 70 | + include_directories(SYSTEM ${Snappy_INCLUDE_DIR}) |
| 71 | + list(APPEND Caffe_LINKER_LIBS ${Snappy_LIBRARIES}) |
| 72 | +endif() |
| 73 | + |
| 74 | +# ---[ CUDA |
| 75 | +include(cmake/Cuda.cmake) |
| 76 | +if(NOT HAVE_CUDA) |
| 77 | + if(CPU_ONLY) |
| 78 | + message(STATUS "-- CUDA is disabled. Building without it...") |
| 79 | + else() |
| 80 | + message(WARNING "-- CUDA is not detected by cmake. Building without it...") |
| 81 | + endif() |
| 82 | + |
| 83 | + # TODO: remove this not cross platform define in future. Use caffe_config.h instead. |
| 84 | + add_definitions(-DCPU_ONLY) |
| 85 | +endif() |
| 86 | + |
| 87 | +if(USE_NCCL) |
| 88 | + include("cmake/External/nccl.cmake") |
| 89 | + include_directories(SYSTEM ${NCCL_INCLUDE_DIR}) |
| 90 | + list(APPEND Caffe_LINKER_LIBS ${NCCL_LIBRARIES}) |
| 91 | + add_definitions(-DUSE_NCCL) |
| 92 | +endif() |
| 93 | + |
| 94 | +# ---[ OpenCV |
| 95 | +if(USE_OPENCV) |
| 96 | + find_package(OpenCV QUIET COMPONENTS core highgui imgproc imgcodecs) |
| 97 | + if(NOT OpenCV_FOUND) # if not OpenCV 3.x, then imgcodecs are not found |
| 98 | + find_package(OpenCV REQUIRED COMPONENTS core highgui imgproc) |
| 99 | + endif() |
| 100 | + include_directories(SYSTEM ${OpenCV_INCLUDE_DIRS}) |
| 101 | + list(APPEND Caffe_LINKER_LIBS ${OpenCV_LIBS}) |
| 102 | + message(STATUS "OpenCV found (${OpenCV_CONFIG_PATH})") |
| 103 | + add_definitions(-DUSE_OPENCV) |
| 104 | +endif() |
| 105 | + |
| 106 | +# ---[ BLAS |
| 107 | +if(NOT APPLE) |
| 108 | + set(BLAS "Atlas" CACHE STRING "Selected BLAS library") |
| 109 | + set_property(CACHE BLAS PROPERTY STRINGS "Atlas;Open;MKL") |
| 110 | + |
| 111 | + if(BLAS STREQUAL "Atlas" OR BLAS STREQUAL "atlas") |
| 112 | + find_package(Atlas REQUIRED) |
| 113 | + include_directories(SYSTEM ${Atlas_INCLUDE_DIR}) |
| 114 | + list(APPEND Caffe_LINKER_LIBS ${Atlas_LIBRARIES}) |
| 115 | + elseif(BLAS STREQUAL "Open" OR BLAS STREQUAL "open") |
| 116 | + find_package(OpenBLAS REQUIRED) |
| 117 | + include_directories(SYSTEM ${OpenBLAS_INCLUDE_DIR}) |
| 118 | + list(APPEND Caffe_LINKER_LIBS ${OpenBLAS_LIB}) |
| 119 | + elseif(BLAS STREQUAL "MKL" OR BLAS STREQUAL "mkl") |
| 120 | + find_package(MKL REQUIRED) |
| 121 | + include_directories(SYSTEM ${MKL_INCLUDE_DIR}) |
| 122 | + list(APPEND Caffe_LINKER_LIBS ${MKL_LIBRARIES}) |
| 123 | + add_definitions(-DUSE_MKL) |
| 124 | + endif() |
| 125 | +elseif(APPLE) |
| 126 | + find_package(vecLib REQUIRED) |
| 127 | + include_directories(SYSTEM ${vecLib_INCLUDE_DIR}) |
| 128 | + list(APPEND Caffe_LINKER_LIBS ${vecLib_LINKER_LIBS}) |
| 129 | + |
| 130 | + if(VECLIB_FOUND) |
| 131 | + if(NOT vecLib_INCLUDE_DIR MATCHES "^/System/Library/Frameworks/vecLib.framework.*") |
| 132 | + add_definitions(-DUSE_ACCELERATE) |
| 133 | + endif() |
| 134 | + endif() |
| 135 | +endif() |
| 136 | + |
| 137 | +# ---[ Python |
| 138 | +if(BUILD_python) |
| 139 | + if(NOT "${python_version}" VERSION_LESS "3.0.0") |
| 140 | + # use python3 |
| 141 | + find_package(PythonInterp 3.0) |
| 142 | + find_package(PythonLibs 3.0) |
| 143 | + find_package(NumPy 1.7.1) |
| 144 | + # Find the matching boost python implementation |
| 145 | + set(version ${PYTHONLIBS_VERSION_STRING}) |
| 146 | + |
| 147 | + STRING( REGEX REPLACE "[^0-9]" "" boost_py_version ${version} ) |
| 148 | + find_package(Boost 1.46 COMPONENTS "python-py${boost_py_version}") |
| 149 | + set(Boost_PYTHON_FOUND ${Boost_PYTHON-PY${boost_py_version}_FOUND}) |
| 150 | + |
| 151 | + while(NOT "${version}" STREQUAL "" AND NOT Boost_PYTHON_FOUND) |
| 152 | + STRING( REGEX REPLACE "([0-9.]+).[0-9]+" "\\1" version ${version} ) |
| 153 | + |
| 154 | + STRING( REGEX REPLACE "[^0-9]" "" boost_py_version ${version} ) |
| 155 | + find_package(Boost 1.46 COMPONENTS "python-py${boost_py_version}") |
| 156 | + set(Boost_PYTHON_FOUND ${Boost_PYTHON-PY${boost_py_version}_FOUND}) |
| 157 | + |
| 158 | + STRING( REGEX MATCHALL "([0-9.]+).[0-9]+" has_more_version ${version} ) |
| 159 | + if("${has_more_version}" STREQUAL "") |
| 160 | + break() |
| 161 | + endif() |
| 162 | + endwhile() |
| 163 | + if(NOT Boost_PYTHON_FOUND) |
| 164 | + find_package(Boost 1.46 COMPONENTS python) |
| 165 | + endif() |
| 166 | + else() |
| 167 | + # disable Python 3 search |
| 168 | + find_package(PythonInterp 2.7) |
| 169 | + find_package(PythonLibs 2.7) |
| 170 | + find_package(NumPy 1.7.1) |
| 171 | + find_package(Boost 1.46 COMPONENTS python) |
| 172 | + endif() |
| 173 | + if(PYTHONLIBS_FOUND AND NUMPY_FOUND AND Boost_PYTHON_FOUND) |
| 174 | + set(HAVE_PYTHON TRUE) |
| 175 | + if(Boost_USE_STATIC_LIBS AND MSVC) |
| 176 | + add_definitions(-DBOOST_PYTHON_STATIC_LIB) |
| 177 | + endif() |
| 178 | + if(BUILD_python_layer) |
| 179 | + add_definitions(-DWITH_PYTHON_LAYER) |
| 180 | + include_directories(SYSTEM ${PYTHON_INCLUDE_DIRS} ${NUMPY_INCLUDE_DIR} ${Boost_INCLUDE_DIRS}) |
| 181 | + list(APPEND Caffe_LINKER_LIBS ${PYTHON_LIBRARIES} ${Boost_LIBRARIES}) |
| 182 | + endif() |
| 183 | + endif() |
| 184 | +endif() |
| 185 | + |
| 186 | +# ---[ Matlab |
| 187 | +if(BUILD_matlab) |
| 188 | + if(MSVC) |
| 189 | + find_package(Matlab COMPONENTS MAIN_PROGRAM MX_LIBRARY) |
| 190 | + if(MATLAB_FOUND) |
| 191 | + set(HAVE_MATLAB TRUE) |
| 192 | + endif() |
| 193 | + else() |
| 194 | + find_package(MatlabMex) |
| 195 | + if(MATLABMEX_FOUND) |
| 196 | + set(HAVE_MATLAB TRUE) |
| 197 | + endif() |
| 198 | + endif() |
| 199 | + # sudo apt-get install liboctave-dev |
| 200 | + find_program(Octave_compiler NAMES mkoctfile DOC "Octave C++ compiler") |
| 201 | + |
| 202 | + if(HAVE_MATLAB AND Octave_compiler) |
| 203 | + set(Matlab_build_mex_using "Matlab" CACHE STRING "Select Matlab or Octave if both detected") |
| 204 | + set_property(CACHE Matlab_build_mex_using PROPERTY STRINGS "Matlab;Octave") |
| 205 | + endif() |
| 206 | +endif() |
| 207 | + |
| 208 | +# ---[ Doxygen |
| 209 | +if(BUILD_docs) |
| 210 | + find_package(Doxygen) |
| 211 | +endif() |
0 commit comments