Sample Distributed TfJob

sample-distributed-tfjob.json

{
  "apiVersion": "tensorflow.org/v1alpha1",
  "kind": "TfJob",
  "metadata": {
    "name": "distributed-job"
  },
  "spec": {
    "replicaSpecs": [
      {
        "replicas": 1,
        "tfReplicaType": "MASTER",
        "template": {
          "spec": {
            "containers": [
              {
                "image": "gcr.io/tf-on-k8s-dogfood/tf_sample:dc944ff",
                "name": "tensorflow"
              }
            ],
            "restartPolicy": "OnFailure"
          }
        }
      },
      {
        "replicas": 1,
        "tfReplicaType": "WORKER",
        "template": {
          "spec": {
            "containers": [
              {
                "image": "gcr.io/tf-on-k8s-dogfood/tf_sample:dc944ff",
                "name": "tensorflow"
              }
            ],
            "restartPolicy": "OnFailure"
          }
        }
      },
      {
        "replicas": 2,
        "tfReplicaType": "PS"
      }
    ]
  }
}