python code to check gpu code example Example: check if tensorflow gpu is installed import tensorflow as tf print(tf.test.gpu_device_name())