apiVersion: batch/v1
kind: Job
metadata:
  name: test-ascend-910
spec:
  template:
    spec:
      nodeSelector:
        hpc.lcpu.dev/partition: npu
      containers:
        - image: crmirror.lcpu.dev/hpcgame/ascend:ubuntu20.04 # Inference image name
          imagePullPolicy: IfNotPresent
          name: test-ascend
          command: # Need to run `source /usr/local/Ascend/ascend-toolkit/set_env.sh` before using pytorch
            - "/bin/bash"
            - "-c"
            - |
              sleep inf
          resources:
            requests:
              cpu: "8"
              memory: 32Gi
              huawei.com/Ascend910: "1"
            limits:
              cpu: "8"
              memory: 32Gi
              huawei.com/Ascend910: "1"
          volumeMounts:
            - name: hiai-driver
              mountPath: /usr/local/Ascend/driver
              readOnly: true
            - name: ascend-install-info
              mountPath: /etc/ascend_install.info
              readOnly: true
            # Your own PVC
            # - name: my-pvc
            #   mountPath: /data
      volumes:
        - name: hiai-driver
          hostPath:
            path: /usr/local/Ascend/driver
        - name: ascend-install-info
          hostPath:
            path: /etc/ascend_install.info
        # Your own PVC
        # - name: my-pvc
        #   persistentVolumeClaim:
        #     claimName: my-pvc
      restartPolicy: OnFailure