blob: de92cd16c612fe1d2f30ed59c889bc3b9de08126 (
plain)
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
39
40
41
|
_BLAS_INCLUDE := $(shell pkgconf --cflags-only-I openblas | sed 's/-I//g')
_NUMPY_INCLUDE := $(shell python -c 'import numpy; print(numpy.get_include())')
_OCVMAJ := $(shell opencv_version | awk -F'.' '{ print $$1 }')
_PYINC := $(shell python -c 'import sysconfig; print(sysconfig.get_path("include"))')
_PYMAJ := $(shell python -c 'import sys; print("%s" %sys.version_info[0])')
_PYVER := $(shell python -c 'import sys; print("%s.%s" %sys.version_info[0:2])')
USE_CUDNN := 0
USE_OPENCV := 1
USE_LEVELDB := 1
USE_LMDB := 1
OPENCV_VERSION := $(_OCVMAJ)
CUSTOM_CXX := $(NVCC_CCBIN)
CUDA_DIR := $(CUDA_PATH)
CUDA_ARCH := -gencode arch=compute_75,code=sm_75 \
-gencode arch=compute_80,code=sm_80 \
-gencode arch=compute_86,code=sm_86 \
-gencode arch=compute_87,code=sm_87 \
-gencode arch=compute_89,code=sm_89 \
-gencode arch=compute_90,code=sm_90 \
-gencode arch=compute_100,code=sm_100 \
-gencode arch=compute_103,code=sm_103 \
-gencode arch=compute_110,code=sm_110 \
-gencode arch=compute_120,code=sm_120 \
-gencode arch=compute_121,code=sm_121 \
-gencode arch=compute_121,code=compute_121
BLAS := open
BLAS_INCLUDE := $(_BLAS_INCLUDE)
BLAS_LIB := /usr/lib
PYTHON_LIBRARIES := boost_python$(_PYMAJ) python$(_PYVER)
PYTHON_INCLUDE := $(_PYINC) $(_NUMPY_INCLUDE)
PYTHON_LIB := /usr/lib
WITH_PYTHON_LAYER := 1
INCLUDE_DIRS := $(PYTHON_INCLUDE)
LIBRARY_DIRS := $(PYTHON_LIB)
USE_NCCL := 1
USE_PKG_CONFIG := 1
BUILD_DIR := build
DISTRIBUTE_DIR := distribute
TEST_GPUID := 0
Q ?= @
|