From 51fbf0127726a659862c9ee3b6b7257fc0a1b39d Mon Sep 17 00:00:00 2001 From: Philip Kahn Date: Fri, 4 May 2018 15:07:45 -0700 Subject: [PATCH] Address GPU message in root dir --- darknet.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/darknet.py b/darknet.py index f8c6401f..873dab74 100644 --- a/darknet.py +++ b/darknet.py @@ -97,26 +97,26 @@ if os.name == "nt": if int(os.environ['CUDA_VISIBLE_DEVICES']) < 0: raise ValueError("ForceCPU") try: - # Check a global global DARKNET_FORCE_CPU if DARKNET_FORCE_CPU: raise ValueError("ForceCPU") except NameError: pass - #print(os.environ.keys()) - #print("FORCE_CPU flag undefined, proceeding with GPU") + # print(os.environ.keys()) + # print("FORCE_CPU flag undefined, proceeding with GPU") if not os.path.exists(winGPUdll): raise ValueError("NoDLL") lib = CDLL(winGPUdll, RTLD_GLOBAL) except (KeyError, ValueError): - print("Notice: GPU-free mode") hasGPU = False if os.path.exists(winNoGPUdll): lib = CDLL(winNoGPUdll, RTLD_GLOBAL) + print("Notice: CPU-only mode") else: # Try the other way, in case no_gpu was # compile but not renamed lib = CDLL(winGPUdll, RTLD_GLOBAL) + print("Environment variables indicated a CPU run, but we didn't find `"+winNoGPUdll+"`. Trying a GPU run anyway.") else: lib = CDLL("./libdarknet.so", RTLD_GLOBAL) lib.network_width.argtypes = [c_void_p]