github.com/kubeflow/training-operator@v1.7.0/examples/pytorch/smoke-dist/pytorch_job_sendrecv.yaml (about)

     1  apiVersion: "kubeflow.org/v1"
     2  kind: "PyTorchJob"
     3  metadata:
     4    name: "pytorch-dist-basic-sendrecv"
     5  spec:
     6    pytorchReplicaSpecs:
     7      Master:
     8        replicas: 1
     9        restartPolicy: OnFailure
    10        template:
    11          spec:
    12            containers:
    13              - name: pytorch
    14                image: kubeflow/pytorch-dist-sendrecv-test:latest
    15      Worker:
    16        replicas: 3
    17        restartPolicy: OnFailure
    18        template:
    19          spec:
    20            containers:
    21              - name: pytorch
    22                image: kubeflow/pytorch-dist-sendrecv-test:latest