tf.distribute.cluster_resolver.GCEClusterResolver

ClusterResolver for Google Compute Engine.

Inherits From: ClusterResolver

This is an implementation of cluster resolvers for the Google Compute Engine instance group platform. By specifying a project, zone, and instance group, this will retrieve the IP address of all the instances within the instance group and return a ClusterResolver object suitable for use for distributed TensorFlow.

Usage example with tf.distribute.Strategy:

  # On worker 0
  cluster_resolver = GCEClusterResolver("my-project", "us-west1",
                                        "my-instance-group",
                                        task_type="worker", task_id=0)
  strategy = tf.distr