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