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 的版本)。