Not creating XLA devices, tf_xla_enable_xla_devices not set
import os
os.environ['TF_XLA_FLAGS'] = '--tf_xla_enable_xla_devices'
could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)