CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix force_plaidML

Browse Source
This commit is contained in:
iperov
2019-03-13 18:30:40 +04:00
parent 58763756f5
commit 90a7d4b1e7
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
nnlib/device.py
View File

@@ -317,7 +317,7 @@ if device.backend is None and not force_tf_cpu:
device.backend = None
has_nvml = False
if not has_nvidia_device and (device.backend is None or force_plaidML):
if force_plaidML or (device.backend is None and not has_nvidia_device):
#tensorflow backend was failed without has_nvidia_device , or forcing plaidML, trying to use plaidML backend
if plaidML_devices_count == 0:
#print ("plaidML: No capable OpenCL devices found. Falling back to tensorflow backend.")
@@ -343,3 +343,4 @@ if device.backend is None:
else:
#has NVSMI, no capable CUDA-devices, also plaidML was failed, then CPU only
device.backend = "tensorflow-cpu"