AttributeError: le module 'tensorflow_core._api.v2.config' n'a pas d'attribut 'list_physical_devices'

Question

J'utilise Tensorflow 2.0 sur Ubuntu 18.04. En cours d'exécution

tf.config.list_physical_devices('GPU')

J'obtiens l'erreur ci-dessus. Quelle est la solution pour cela?

Answer 1

Exécutez tf.config.experimental.list_physical_devices('GPU') pour voir tous les GPU

ou

exécutez tf.config.experimental.list_physical_devices(device_type=None) pour voir tous les périphériques

Vous pouvez également utiliser tf.test.is_gpu_available(). Il dit True s'il détecte le GPU disponible.