jboscher personnel demandée il y a 4 ans
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
