import tensorflow as tf
tf.debugging.set_log_device_placement(True)
import logging

# Create some tensors
a = tf.constant([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])
b = tf.constant([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0]])
c = tf.matmul(a, b)

print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
print("GPU used for operation : ", c)
logging.critical("Reached end")