github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-15 01:23:44 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

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"