Check whether Tensorflow is running on GPU

Question

I have read many questions and "guides" on how to understand if Tensorflow is running on GPU but I am still quite confused.

I have taken a screenshot of my session and I would like to understand what is going on, and if Tensorflow is running on GPU or CPU.

Answer 1

Roberto, try this

import tensorflow as tf
if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
    print("Please install GPU version of TF")

the output should be

Default GPU Device: /device:GPU:0

GPU 0 is your GTX 860m