Список устройств в TensorFlow
Чтобы узнать, какие устройства
доступны процессу, служит функция
list_physical_devices.
Без аргумента она возвращает
все найденные устройства:
import tensorflow as tf
devices = tf.config.list_physical_devices()
print(devices)
# выведет [PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]
Тот же запрос можно сузить до процессоров или видеокарт, передав строку с типом устройства:
import tensorflow as tf
cpus = tf.config.list_physical_devices('CPU')
gpus = tf.config.list_physical_devices('GPU')
print(cpus)
print(gpus)
# выведет [PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]
# выведет []
Выведите полный перечень физических устройств, которые видит TensorFlow в этом процессе.
Получите отдельно список процессоров и отдельно список видеокарт и выведите оба результата подряд.
Сохраните число найденных видеокарт в переменную и выведите это число.
