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

OpenCL support: fix bug, adding new requirements files

Browse source
This commit is contained in:
iperov 2019-02-19 20:48:39 +04:00
parent 72ba6b103c
commit b939fe9d85
4 changed files with 18 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
nnlib/device.py
View file

@ -270,6 +270,8 @@ try:
ctx = plaidml.Context()
for d in plaidml.devices(ctx, return_all=True)[0]:
details = json.loads(d.details)
if details['type'] == 'CPU': #skipping opencl-CPU
continue
if 'nvidia' in details['vendor'].lower():
has_nvidia_device = True
plaidML_devices += [ {'id':d.id,