darknet.py - apply the same changes to build/darknet/x64/

pull/856/head
Vinjn Zhang 7 years ago committed by GitHub
parent dbcdac88a5
commit c6ca54b411
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 7
      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"]:

Loading…
Cancel
Save