github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-07 05:22:06 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix for plaidml

Browse source
This commit is contained in:
Colombo 2019-10-08 16:55:21 +04:00
parent ac7725163d
commit 627df082d7
2 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
nnlib/device.py
View file

@ -232,6 +232,7 @@ class device:
plaidML_build = os.environ.get("DFL_PLAIDML_BUILD", "0") == "1"
plaidML_devices = None
plaidML_devices_count = 0
cuda_devices = None
if plaidML_build:
@ -253,8 +254,8 @@ if plaidML_build:
ctx.shutdown()
except:
pass
if len(plaidML_devices) != 0:
plaidML_devices_count = len(plaidML_devices)
if plaidML_devices_count != 0:
device.backend = "plaidML"
else:
if cuda_devices is None: