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': Вычислительная способность устройства CUDA в виде кортежа из двух целых чисел, в формате (major_version, minor_version). Доступно только для графических процессоров NVIDIA.

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