sigs.k8s.io/kueue@v0.6.2/site/static/examples/jobs/sample-tfjob.yaml (about)

     1  apiVersion: kubeflow.org/v1
     2  kind: TFJob
     3  metadata:
     4    name: tensorflow-dist-mnist
     5    namespace: default
     6    labels:
     7      kueue.x-k8s.io/queue-name: user-queue
     8  spec:
     9    tfReplicaSpecs:
    10      PS:
    11        replicas: 1
    12        restartPolicy: Never
    13        template:
    14          spec:
    15            containers:
    16              - name: tensorflow
    17                image: kubeflow/tf-dist-mnist-test:latest
    18                resources:
    19                  requests:
    20                    cpu: 1
    21                    memory: "200Mi"
    22      Worker:
    23        replicas: 2
    24        restartPolicy: Never
    25        template:
    26          spec:
    27            containers:
    28              - name: tensorflow
    29                image: kubeflow/tf-dist-mnist-test:latest
    30                resources:
    31                  requests:
    32                    cpu: 1
    33                    memory: "200Mi"