Python tf.device用法及代码示例

指定在此上下文中创建/执行的操作的设备。

用法

tf.device(
    device_name
)

此函数指定用于在特定上下文中创建/执行的操作的设备。嵌套上下文将继承并在指定设备上创建/执行其操作。如果不需要特定设备，请考虑不使用此函数，以便自动分配设备。通常，此函数的使用是可选的。 device_name 可以完全指定，如“/job:worker/task:1/device:cpu:0”，或部分指定，仅包含"/" 分隔字段的子集。指定的任何字段都将覆盖来自外部范围的设备注释。

例如：

with tf.device('/job:foo'):
  # ops created here have devices with /job:foo
  with tf.device('/job:bar/task:0/device:gpu:2'):
    # ops created here have the fully specified device above
  with tf.device('/device:gpu:1'):
    # ops created here have the device '/job:foo/device:gpu:1'