Python tf.config.experimental.get_device_details用法及代码示例

返回有关物理设备的详细信息。

用法

tf.config.experimental.get_device_details(
    device
)

此 API 接受 tf.config.list_physical_devices 返回的 tf.config.PhysicalDevice 。它返回一个带有字符串键的字典，其中包含有关设备的各种详细信息。每个键仅由设备子集支持，因此您不应假设返回的 dict 将具有任何特定键。

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

目前，仅返回 GPU 的详细信息。如果传递一个非 GPU 设备，此函数返回一个空字典。

返回的 dict 可能有以下键：

• 'device_name' ：设备的人类可读名称作为字符串，例如"Titan V"。与 tf.config.PhysicalDevice.name 不同，如果每个设备是相同的型号，这对于多个设备都是相同的。目前仅适用于 GPU。
• 'compute_capability' ：设备的计算能力作为两个整数的元组，格式为 (major_version, minor_version) 。仅适用于 NVIDIA GPU

注意：这与tf.sysconfig.get_build_info 相似，因为这两个函数都可以返回与 GPU 相关的信息。但是，此函数返回有关特定设备的运行时信息(例如 GPU 的计算能力)，而 tf.sysconfig.get_build_info 返回有关如何构建 TensorFlow 的编译时信息(例如构建 CUDA TensorFlow 的版本)。