From 9ff0b7efcb9281bd68ffc9b670e4f882a03dc559 Mon Sep 17 00:00:00 2001
From: AlexeyAB <kikots@mail.ru>
Date: Thu, 03 Aug 2017 19:18:00 +0000
Subject: [PATCH] Linux supported

---
 Makefile |   15 ++++++++++++++-
 1 files changed, 14 insertions(+), 1 deletions(-)

diff --git a/Makefile b/Makefile
index f5524b9..a2a90f7 100644
--- a/Makefile
+++ b/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

--
Gitblit v1.10.0