在创建POD时,YAML中容器请求带上资源类型 nvidia.com/gpu,
可以将POD调度到GPU节点上 。
apiVersion: v1
kind: Pod
metadata: name: gpu-pod
spec:
containers:
- name: cuda-container
image: nvidia/cuda:10.0-base
resources:
limits: nvidia.com/gpu: 1