To find available devices in TensorFlow, from Python prompt use –
import tensorflow as tf
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
If you want to find out which device is being used, try –
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
Source: Stackoverflow