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).