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：設備上可用的最大內存量。注意：根據設備的不同，可用內存可能會大大減少。