Update Makefile

pjreddie#289 (comment)
ashirgao · Mar 8, 2019 · 68560d4 · 68560d4
1 parent bf54732
commit 68560d4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/Makefile b/Makefile
@@ -1,8 +1,9 @@
-GPU=0
-CUDNN=0
-OPENCV=0
+GPU=1
+CUDNN=1
+OPENCV=1
 OPENMP=0
 DEBUG=0
+NUMPY=1
 
 ARCH= -gencode arch=compute_30,code=sm_30 \
       -gencode arch=compute_35,code=sm_35 \
@@ -46,6 +47,11 @@ LDFLAGS+= `pkg-config --libs opencv` -lstdc++
 COMMON+= `pkg-config --cflags opencv` 
 endif
 
+ifeq ($(NUMPY), 1) 
+COMMON+= -DNUMPY -I/usr/include/python2.7/ -I/usr/lib/python2.7/dist-packages/numpy/core/include/numpy/
+CFLAGS+= -DNUMPY
+endif
+
 ifeq ($(GPU), 1) 
 COMMON+= -DGPU -I/usr/local/cuda/include/
 CFLAGS+= -DGPU