apiVersion: v1
kind: Pod
metadata:
  name: nbody-pod
spec:
  restartPolicy: OnFailure
  containers:
    - name: nbody
      image: "nvcr.io/nvidia/k8s/cuda-sample:nbody"
      command: ["nbody"]
      args: ["-benchmark", "-i=10000000"]
      resources:
        limits:
          nvidia.com/gpu: 1