check tensorflow using cpu or gpu code example Example: check if tensorflow gpu is installed import tensorflow as tf print(tf.test.gpu_device_name())