test-tf.py

import tensorflow as tf

with tf.Session() as sess:
  devices = sess.list_devices()