diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2017-06-30 20:17:24 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2017-06-30 20:17:24 +0200 |
| commit | 7c5ca9fdcafcec0e3781ed856eba42cd608f0e15 (patch) | |
| tree | a4425e8ce03538c1741e6c10b0a2459bcb03aea2 /python | |
| parent | 82cf40b516b5ab11c34b3642a01603ec1b590c9f (diff) | |
es6 stuff
Diffstat (limited to 'python')
| -rw-r--r-- | python/devices.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/python/devices.py b/python/devices.py new file mode 100644 index 0000000..a24be23 --- /dev/null +++ b/python/devices.py @@ -0,0 +1,8 @@ +#!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) |