job-example-gpu.yaml

apiVersion: batch/v1
kind: Job
metadata:
  name: job222234567
  namespace: default
  labels:
    kueue.x-k8s.io/queue-name: user-queue
spec:
  parallelism: 3
  completions: 3
  template:
    spec:
      volumes:
      - name: script
        hostPath:
          path: /nfs/entry.sh
          type: File
      - name: data
        hostPath:
          path: /nfs/home/demo_admin/data
          type: ""
      - name: algorithm
        hostPath:
          path: /nfs/home/demo_admin/algorithm
          type: ""
      - name: workdir
        hostPath:
          path: /nfs/home/demo_admin
          type: ""
      - name: extramount
        hostPath:
          path: /nfs/home/demo_admin
          type: ""
      nodeSelector:
        kubernetes.io/os: linux
      containers:
      - image: jupyter/minimal-notebook:latest
        imagePullPolicy: IfNotPresent
        name: job222234567
        command: ["bash", "/opt/entry.sh", "20000", "k8s-master", "30001"]
        env:
        - name: WORK_DIR
          value: /nfs/home/demo_admin
        resources:
          limits:
            nvidia.com/gpu: 1
        workingDir: /nfs/home/demo_admin
        volumeMounts:
        - mountPath: /opt/entry.sh
          name: script
        - mountPath: /nfs/home/demo_admin/data
          name: data
        - mountPath: /nfs/home/demo_admin/algorithm
          name: algorithm
        - mountPath: /nfs/home/demo_admin
          name: workdir
        - mountPath: /nfs/home/demo_admin
          name: extramount
      restartPolicy: OnFailure