From 51fbf0127726a659862c9ee3b6b7257fc0a1b39d Mon Sep 17 00:00:00 2001
From: Philip Kahn <tigerhawkvok@gmail.com>
Date: Fri, 04 May 2018 22:07:45 +0000
Subject: [PATCH] Address GPU message in root dir
---
darknet.py | 8 ++++----
1 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/darknet.py b/darknet.py
index f8c6401..873dab7 100644
--- a/darknet.py
+++ b/darknet.py
@@ -97,26 +97,26 @@
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]
--
Gitblit v1.10.0