k8s_cuda_gpu_test.yaml

---
apiVersion: v1
kind: Pod
metadata:
  name: gpu-test
  namespace: default
spec:
  tolerations:
    - key: nvidia.com/gpu
      operator: Exists
      effect: NoSchedule
  containers:
    - name: gpu-test
      image: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
      imagePullPolicy: Always
      command: ["/bin/bash", "-c", "--"]
      args: ["while true; do sleep 5; done;"]
      resources:
        limits:
          nvidia.com/gpu: 1