Verifique si se instala TensorFlow GPU
import tensorflow as tf
print(tf.test.gpu_device_name())
CBT fan club
import tensorflow as tf
print(tf.test.gpu_device_name())
tf.config.list_physical_devices('GPU')
$ pip uninstall tensorflow
$ pip install tensorflow-gpu