Проверьте CUDA доступный TensorFlow

print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
XeN0N