code to check if gpu is working code example

Example: check if tensorflow gpu is installed

import tensorflow as tf
print(tf.test.gpu_device_name())