simple code for checking tf gpu code example Example: check if tensorflow gpu is installed import tensorflow as tf print(tf.test.gpu_device_name())