---
apiVersion: apps/v1
kind: DaemonSet
metadata:
  namespace: default
  name: gpu-pod
spec:
  selector:
    matchLabels:
      app: gpu-pod
  template:
    metadata:
      labels:
        app: gpu-pod
    spec:
      nodeSelector:
        nvidia.com/gpu.family: pascal
      containers:
        - name: gpu-pod
          image: nvidia/cuda:12.3.1-runtime-ubuntu20.04
          command: ["sleep", "infinity"]
      terminationGracePeriodSeconds: 0
      runtimeClassName: nvidia
      restartPolicy: Always