Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

tf.test.gpu_device_name

TensorFlow 2 version View source on GitHub

Returns the name of a GPU device if available or the empty string.