将任务分配到指定 GPU

目录

将任务分配到指定 GPU

要将任务分配到指定的 GPU,只需在 annotations 字段中设置 nvidia.com/use-gpuuuid

apiVersion: v1
kind: Pod
metadata:
  name: gpu-pod
  annotations:
    nvidia.com/use-gpuuuid: "GPU-123456"
spec:
  containers:
    - name: ubuntu-container
      image: ubuntu:18.04
      command: ["bash", "-c", "sleep 86400"]
      resources:
        limits:
          nvidia.com/gpu: 2 # requesting 2 vGPUs