nnlib added force_has_nvidia_device

2019-02-25 21:15:58 +04:00
parent 281787ac3e
commit ac5ac75645
1 changed files with 1 additions and 1 deletions
@@ -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