From 07a9f2c2ce084412fffa54e88e6ff01adca734f5 Mon Sep 17 00:00:00 2001
From: scirocco <zyf199966@126.com>
Date: Mon, 17 Feb 2020 12:33:16 +0800
Subject: [PATCH] fix CUDA compile

---
 src/libraries/darknet/CMakeLists.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/libraries/darknet/CMakeLists.txt b/src/libraries/darknet/CMakeLists.txt
index 2388098..352b8cb 100644
--- a/src/libraries/darknet/CMakeLists.txt
+++ b/src/libraries/darknet/CMakeLists.txt
@@ -1,6 +1,7 @@
 cmake_minimum_required(VERSION 3.14)
 project(darknet)
 
+set(CMAKE_CUDA_COMPILER "/usr/local/cuda/bin/nvcc")
 enable_language(C)
 enable_language(CXX)
 enable_language(CUDA)
-- 
Gitee