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

     1  apiVersion: kubeflow.org/v2beta1
     2  kind: MPIJob
     3  metadata:
     4    name: pi
     5    labels:
     6      kueue.x-k8s.io/queue-name: user-queue
     7  spec:
     8    slotsPerWorker: 1
     9    runPolicy:
    10      cleanPodPolicy: Running
    11      ttlSecondsAfterFinished: 60
    12    sshAuthMountPath: /home/mpiuser/.ssh
    13    mpiReplicaSpecs:
    14      Launcher:
    15        replicas: 1
    16        template:
    17          spec:
    18            containers:
    19            - image: mpioperator/mpi-pi:openmpi
    20              name: mpi-launcher
    21              securityContext:
    22                runAsUser: 1000
    23              command:
    24              - mpirun
    25              args:
    26              - -n
    27              - "2"
    28              - /home/mpiuser/pi
    29              resources:
    30                limits:
    31                  cpu: 1
    32                  memory: 1Gi
    33      Worker:
    34        replicas: 2
    35        template:
    36          spec:
    37            containers:
    38            - image: mpioperator/mpi-pi:openmpi
    39              name: mpi-worker
    40              securityContext:
    41                runAsUser: 1000
    42              command:
    43              - /usr/sbin/sshd
    44              args:
    45              - -De
    46              - -f
    47              - /home/mpiuser/.sshd_config
    48              resources:
    49                limits:
    50                  cpu: 1
    51                  memory: 1Gi