forked from BVLC/caffe
-
Notifications
You must be signed in to change notification settings - Fork 8
/
Copy pathcaffe_config.h
38 lines (30 loc) · 820 Bytes
/
caffe_config.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
/* Sources directory */
#define SOURCE_FOLDER "/home/tl486/git_TCcaffe/caffe_solverOriginal/caffe"
/* Binaries directory */
#define BINARY_FOLDER "/home/tl486/git_TCcaffe/caffe_solverOriginal/caffe"
/* NVIDA Cuda */
#define HAVE_CUDA
/* NVIDA cuDNN */
#define HAVE_CUDNN
#define USE_CUDNN
/* NVIDA cuDNN */
/* #undef CPU_ONLY */
/* Test device */
#define CUDA_TEST_DEVICE -1
/* Temporary (TODO: remove) */
#if 1
#define CMAKE_SOURCE_DIR SOURCE_FOLDER "/src/"
#define EXAMPLES_SOURCE_DIR BINARY_FOLDER "/examples/"
#define CMAKE_EXT ".gen.cmake"
#else
#define CMAKE_SOURCE_DIR "src/"
#define EXAMPLES_SOURCE_DIR "examples/"
#define CMAKE_EXT ""
#endif
/* Matlab */
/* #undef HAVE_MATLAB */
/* IO libraries */
/* #undef USE_OPENCV */
#define USE_LEVELDB
#define USE_LMDB
/* #undef ALLOW_LMDB_NOLOCK */