Функция config.experimental.get_device_details

Функция config.experimental.get_device_details возвращает словарь с подробной информацией о переданном физическом устройстве. Первым параметром функция принимает объект устройства, полученный, например, через config.list_physical_devices. Функция полезна, когда нужно узнать имя устройства, его вычислительную способность или тип ускорителя.

Синтаксис

tf.config.experimental.get_device_details(device)

Пример

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

import tensorflow as tf

devices = tf.config.list_physical_devices()
for device in devices:
	details = tf.config.experimental.get_device_details(device)
	print(device.device_type, details)

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

"CPU {'device_name': 'CPU'}"

Пример

Давайте получим детали конкретного CPU-устройства и обратимся к ключу device_name:

import tensorflow as tf

cpus = tf.config.list_physical_devices('CPU')
if cpus:
	details = tf.config.experimental.get_device_details(cpus[0])
	print(details.get('device_name'))

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

"CPU"

Пример

Давайте проверим наличие GPU и выведем информацию о вычислительной способности:

import tensorflow as tf

gpus = tf.config.list_physical_devices('GPU')
for gpu in gpus:
	details = tf.config.experimental.get_device_details(gpu)
	print(details.get('compute_capability'))

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

"8.6"

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

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