Alexey
2017-08-03 de625aa4a7fc5b6b1a7fd1b9a4c13d06b3a8c9a1
Makefile
@@ -2,8 +2,16 @@
CUDNN=0
OPENCV=0
DEBUG=0
OPENMP=0
ARCH= --gpu-architecture=compute_52 --gpu-code=compute_52
ARCH= -gencode arch=compute_20,code=[sm_20,sm_21] \
      -gencode arch=compute_30,code=sm_30 \
      -gencode arch=compute_35,code=sm_35 \
      -gencode arch=compute_50,code=[sm_50,compute_50] \
      -gencode arch=compute_52,code=[sm_52,compute_52]
# This is what I use, uncomment if you know your arch and want to specify
# ARCH=  -gencode arch=compute_52,code=compute_52
VPATH=./src/
EXEC=darknet
@@ -29,6 +37,11 @@
COMMON+= `pkg-config --cflags opencv` 
endif
ifeq ($(OPENMP), 1)
CFLAGS+= -fopenmp
LDFLAGS+= -lgomp
endif
ifeq ($(GPU), 1) 
COMMON+= -DGPU -I/usr/local/cuda/include/
CFLAGS+= -DGPU