mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-10 23:33:30 -07:00
fix for plaidML
This commit is contained in:
parent
8e16f4fa1b
commit
dcbdc79e84
1 changed files with 2 additions and 1 deletions
|
@ -301,7 +301,7 @@ def get_plaidML_devices():
|
||||||
return plaidML_devices
|
return plaidML_devices
|
||||||
|
|
||||||
if not has_nvidia_device:
|
if not has_nvidia_device:
|
||||||
get_plaidML_devices()
|
get_plaidML_devices()
|
||||||
|
|
||||||
#choosing backend
|
#choosing backend
|
||||||
|
|
||||||
|
@ -335,6 +335,7 @@ if force_plaidML or (device.backend is None and not has_nvidia_device):
|
||||||
device.backend = None
|
device.backend = None
|
||||||
else:
|
else:
|
||||||
device.backend = "plaidML"
|
device.backend = "plaidML"
|
||||||
|
plaidML_devices_count = len(get_plaidML_devices())
|
||||||
|
|
||||||
if device.backend is None:
|
if device.backend is None:
|
||||||
if force_tf_cpu:
|
if force_tf_cpu:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue