apiVersion: v1
kind: Pod
metadata:
  name: gpu-test-pod
  namespace: bioinformatics
spec:
  restartPolicy: Never
  nodeSelector:
    gpu: 'yes'
    gpu-type: 'geforce-rtx-3090'
  containers:
  - name: cuda-test
    # Using a standard, lightweight NVIDIA CUDA image to test the runtime
    image: nvidia/cuda:11.8.0-base-ubuntu22.04
    command: ["nvidia-smi"]
    resources:
      limits:
        nvidia.com/gpu: 1