From ac5ac7564540d877f3b3b556a270a3608258b8b6 Mon Sep 17 00:00:00 2001 From: iperov Date: Mon, 25 Feb 2019 21:15:58 +0400 Subject: [PATCH] nnlib added force_has_nvidia_device --- nnlib/device.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nnlib/device.py b/nnlib/device.py index a8f6c72..418d769 100644 --- a/nnlib/device.py +++ b/nnlib/device.py @@ -260,7 +260,7 @@ class device: force_plaidML = os.environ.get("force_plaidML", "0") == "1" has_nvml = False has_nvml_cap = False -has_nvidia_device = False +has_nvidia_device = os.environ.get("force_has_nvidia_device", "0") == "1" plaidML_devices = [] # Using plaidML OpenCL backend to determine system devices and has_nvidia_device