Функция config.list_physical_devices

Функция config.list_physical_devices возвращает список физических устройств, которые видит TensorFlow на текущей машине. Первым параметром передается тип устройства в виде строки, например 'CPU' или 'GPU'. Если параметр не указан, функция возвращает все доступные физические устройства.

Каждый элемент списка представляет собой объект PhysicalDevice с двумя атрибутами: name - имя устройства (например, '/physical_device:CPU:0') и device_type - тип устройства ('CPU' или 'GPU'). Функция полезна для определения доступного оборудования перед запуском вычислений.

Синтаксис

tf.config.list_physical_devices(device_type=None)

Пример

Давайте получим список всех физических устройств, доступных TensorFlow:

import tensorflow as tf

devices = tf.config.list_physical_devices()
print(devices)

Результат выполнения кода:

[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]

Пример

Давайте отфильтруем устройства только по типу 'CPU':

import tensorflow as tf

cpus = tf.config.list_physical_devices('CPU')
print(cpus)

Результат выполнения кода:

[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]

Пример

Давайте проверим, есть ли на машине GPU, и выведем количество найденных устройств:

import tensorflow as tf

gpus = tf.config.list_physical_devices('GPU')
print('GPUs found:', len(gpus))
for gpu in gpus:
	print(gpu.name)

Результат выполнения кода:

"GPUs found: 0"

Смотрите также

функцию config.list_logical_devices,
которая возвращает список логических устройств
функцию config.set_visible_devices,
которая задает видимые устройства для TensorFlow
функцию config.get_visible_devices,
которая возвращает видимые устройства
функцию experimental.get_device_details,
которая возвращает подробную информацию об устройстве