Распечатайте весь графический процессор доступен тензор
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
Better Barracuda