path: root/python/devices.py

#!python

from tensorflow.python.client import device_lib
import json

device_ids = [device.name for device in device_lib.list_local_devices()]

print json.dumps(device_ids)