Aller à la barre d’outils

Comment compter les GPU actifs avec Tf 2.0

DWQA QuestionsComment compter les GPU actifs avec Tf 2.0
jboscher personnel demandée il y a 6 mois

Pour savoir si on a des GPU actifs :
print(tf.test.is_gpu_available())

Pour la liste des GPU :
tf.config.list_physical_devices(‘GPU‘)

S’il y en a un seul, la réponse sera :
[PhysicalDevice(name=’/physical_device:GPU:0′, device_type=’GPU’)]

et pour :
tf.config.experimental.list_physical_devices(device_type=None)
on aura (par exemple)
[PhysicalDevice(name=’/physical_device:CPU:0′, device_type=’CPU’),
PhysicalDevice(name=’/physical_device:XLA_CPU:0′, device_type=’XLA_CPU’),
PhysicalDevice(name=’/physical_device:XLA_GPU:0′, device_type=’XLA_GPU’),
PhysicalDevice(name=’/physical_device:GPU:0′, device_type=’GPU’)]

Pour s’assurer qu’il y a au moins un GPU :
physical_devices = tf.config.experimental.list_physical_devices(‘GPU’)
assert len(physical_devices) > 0, « No GPUs found. »

Evidemment on peut aussi tester les CPUs