Verifique o CUDA disponível Tensorflow
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
XeN0N
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))