tf.config.experimental.get_device_details

Возвращает подробную информацию о физическом устройстве.

tf.config.experimental.get_device_details(
    device
)

Этот API принимает tf.config.PhysicalDevice, возвращаемый функцией tf.config.list_physical_devices. Он возвращает словарь со строковыми ключами, содержащими различные детали устройства. Каждый ключ поддерживается только подмножеством устройств, поэтому не следует предполагать, что возвращаемый словарь будет содержать какой-либо конкретный ключ.

gpu_devices = tf.config.list_physical_devices('GPU')
if gpu_devices:
  details = tf.config.experimental.get_device_details(gpu_devices[0])
  details.get('device_name', 'Unknown GPU')

В настоящее время подробная информация возвращается только для графических процессоров. Если передать устройство, которое не является графическим процессором, эта функция возвращает пустой словарь.

Возвращаемый словарь может содержать следующие ключи:

• 'device_name': Человекопонятное имя устройства в виде строки, например, "Titan V". В отличие от tf.config.PhysicalDevice.name, это имя будет одинаковым для нескольких устройств, если каждое устройство — одной модели. В настоящее время доступно только для графических процессоров.
• 'compute_capability': Вычислительная способность устройства в виде кортежа из двух целых чисел, в формате (major_version, minor_version). Доступно только для графических процессоров NVIDIA

Примечание: Это аналогично tf.sysconfig.get_build_info, так как обе функции могут возвращать информацию, относящуюся к графическим процессорам. Однако эта функция возвращает информацию о работе устройства в реальном времени (например, вычислительную способность графического процессора), в то время как tf.sysconfig.get_build_info возвращает информацию о времени компиляции о том, как был скомпилирован TensorFlow (например, о том, какая версия CUDA использовалась для компиляции TensorFlow).