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