david/DeepFaceLab_Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nnlib added force_has_nvidia_device

Browse Source
This commit is contained in:
iperov
2019-02-25 21:15:58 +04:00
parent 281787ac3e
commit ac5ac75645
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nnlib/device.py
+1 -1
View File
@@ -260,7 +260,7 @@ class device:
force_plaidML = os.environ.get("force_plaidML", "0") == "1" force_plaidML = os.environ.get("force_plaidML", "0") == "1"
has_nvml = False has_nvml = False
has_nvml_cap = False has_nvml_cap = False
has_nvidia_device = False has_nvidia_device = os.environ.get("force_has_nvidia_device", "0") == "1"
plaidML_devices = [] plaidML_devices = []
# Using plaidML OpenCL backend to determine system devices and has_nvidia_device # Using plaidML OpenCL backend to determine system devices and has_nvidia_device