Функция config.get_visible_devices
Функция config.get_visible_devices возвращает
список устройств, которые в данный момент видимы
для TensorFlow. Первым параметром функция принимает
тип устройства в виде строки, например 'GPU'
или 'CPU'. Вторым необязательным параметром
можно передать логическое значение, указывающее,
нужно ли возвращать логические устройства вместо
физических. Если параметр не указан, возвращаются
физические устройства.
Синтаксис
tf.config.get_visible_devices(device_type=None, logical=False)
Пример
Давайте получим список всех видимых физических устройств, не указывая их тип:
import tensorflow as tf
devices = tf.config.get_visible_devices()
print(devices)
Результат выполнения кода:
[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]
Пример
Давайте получим список видимых устройств
только типа 'CPU':
import tensorflow as tf
cpus = tf.config.get_visible_devices('CPU')
print(cpus)
Результат выполнения кода:
[PhysicalDevice(name='/physical_device:CPU:0', device_type='CPU')]
Пример
Давайте получим список видимых логических
устройств типа 'CPU', передав второй
параметр logical=True:
import tensorflow as tf
logical_cpus = tf.config.get_visible_devices('CPU', logical=True)
print(logical_cpus)
Результат выполнения кода:
[LogicalDevice(name='/device:CPU:0', device_type='CPU')]
Смотрите также
-
функцию
config.list_physical_devices,
которая возвращает список физических устройств -
функцию
config.list_logical_devices,
которая возвращает список логических устройств -
функцию
config.set_visible_devices,
которая задает список видимых устройств -
функцию
device,
которая создает контекст для выполнения операций на устройстве
