Python tf.config.experimental.ClusterDeviceFilters用法及代码示例

表示集群中远程工作人员的设备过滤器集合。

用法

tf.config.experimental.ClusterDeviceFilters()

注意：这是一个实验性 API，可能会发生变化。

为选择性作业和任务设置设备过滤器。对于每个远程工作人员，设备过滤器是一个字符串列表。当存在任何过滤器时，远程工作人员将忽略与其任何过滤器不匹配的所有设备。每个过滤器都可以部分指定，例如"/job:ps"、"/job:worker/replica:3" 等。请注意，设备对其所在的工作人员始终可见。

例如，为参数服务器集群设置设备过滤器：

cdf = tf.config.experimental.ClusterDeviceFilters()
for i in range(num_workers):
  cdf.set_device_filters('worker', i, ['/job:ps'])
for i in range(num_ps):
  cdf.set_device_filters('ps', i, ['/job:worker'])

tf.config.experimental_connect_to_cluster(cluster_def,
                                          cluster_device_filters=cdf)

可以部分指定设备过滤器。对于没有指定设备过滤器的远程任务，所有设备都对其可见。