Python tf.compat.v1.Session.list_devices用法及代码示例

用法

list_devices()

列出此会话中的可用设备。

devices = sess.list_devices()
for d in devices:
  print(d.name)

在哪里：

列表中的每个元素都具有以下属性

• name：带有设备全名的字符串。前任：/job:worker/replica:0/task:3/device:CPU:0
• device_type：设备的类型(例如CPU,GPU,TPU.)
• memory_limit：设备上可用的最大内存量。注意：根据设备的不同，可用内存可能会大大减少。