From c6ca54b411478a8bcf3ec8bc7615e80d63bdaba7 Mon Sep 17 00:00:00 2001 From: Vinjn Zhang Date: Sun, 20 May 2018 21:12:35 +0800 Subject: [PATCH] darknet.py - apply the same changes to build/darknet/x64/ --- build/darknet/x64/darknet.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/build/darknet/x64/darknet.py b/build/darknet/x64/darknet.py index b290a358..75ef04b5 100644 --- a/build/darknet/x64/darknet.py +++ b/build/darknet/x64/darknet.py @@ -78,13 +78,14 @@ class METADATA(Structure): #lib = CDLL("darknet.so", RTLD_GLOBAL) hasGPU = True if os.name == "nt": - winGPUdll = "yolo_cpp_dll.dll" - winNoGPUdll = "yolo_cpp_dll_nogpu.dll" + cwd = os.path.dirname(__file__) + os.environ['PATH'] = cwd + ';' + os.environ['PATH'] + winGPUdll = os.path.join(cwd, "yolo_cpp_dll.dll") + winNoGPUdll = os.path.join(cwd, "yolo_cpp_dll_nogpu.dll") envKeys = list() for k, v in os.environ.items(): envKeys.append(k) try: - tmp = os.environ["CUDA_HOME"] try: tmp = os.environ["FORCE_CPU"].lower() if tmp in ["1", "true", "yes", "on"]: