nvidia/k8s-device-plugin 多GPU 设备选择访问

最新推荐文章于 2024-08-12 21:09:21 发布

拜思特

最新推荐文章于 2024-08-12 21:09:21 发布

阅读量152

点赞数 1

文章标签： kubernetes 容器云原生 gpu算力

本文链接：https://blog.csdn.net/qq836825331/article/details/140990111

版权

在一机多卡的节点上，如果只想要pod调度到指定的gpu设备上，可以通过NVIDIA_VISIBLE_DEVICES进行设置，如果不设置默认为 all

apiVersion: v1
kind: Pod
metadata:
  name: ndp-gpu-pod-test
  namespace: default
spec:
  containers:
    - name: ubuntu-container
      image: ubuntu:18.04
      command: [ "bash", "-c", "sleep 86400" ]
      env:
        - name: NVIDIA_VISIBLE_DEVICES #设置可访问的gpu设备
          value: "0,1" #第几个，也可以指定UUID,可指定多个
      resources:
        limits:
          nvidia.com/gpu: 1

  tolerations: #GPU节点污点容忍
    - key: "gpu"
      operator: "Equal"
      value: "on"
      effect: "NoSchedule