TensorFlow 列表设备

在 TensorFlow 中，可以通过 tf.device() 函数将操作分配到不同的设备上运行。此外，可以使用 tf.config.list_physical_devices() 函数列出可用的物理设备。

列出可用设备

import tensorflow as tf

physical_devices = tf.config.list_physical_devices()
print("可用设备：", physical_devices)

列出可用设备的函数为 tf.config.list_physical_devices()，该函数返回一个列表，其中包含每个可用的物理设备。

指定设备运行操作

import tensorflow as tf

with tf.device('/CPU:0'):
    # 将操作分配到 CPU 设备上
    a = tf.constant([1, 2, 3])
    b = tf.constant([4, 5, 6])
    c = tf.add(a, b)

print(c)

使用 with tf.device() 语句，可以将操作分配到指定的设备上运行。在上面的示例中，a 和 b 两个常量张量将被分配到 /CPU:0 设备上运行，c 张量则被添加到默认设备上。如果想将所有操作都分配到 /CPU:0 设备上运行，可以在 with 语句前使用 tf.device('/CPU:0') 语句。

参考文献

• TensorFlow 列表设备官方文档