跳转至

使用 MIG GPU 资源

本节介绍应用如何使用 MIG GPU 资源。

前提条件

界面使用 MIG GPU

  1. 确认集群是否已识别 GPU 卡类型

    进入集群详情 -> 集群设置 -> Addon 设置,查看是否已正确识别,自动识别频率为 10 分钟

    gpu

  2. 通过镜像部署应用可选择并使用 Nvidia MIG 资源。

    mig02

YAML 配置使用 MIG

  1. MIG Single 模式:
apiVersion: apps/v1
kind: Deployment
metadata:
  name: mig-demo
  namespace: default
spec:
  replicas: 1
  selector:
    matchLabels:
      app: mig-demo
  template:
    metadata:
      creationTimestamp: null
      labels:
        app: mig-demo
    spec:
      containers:
        - name: mig-demo1
          image: chrstnhntschl/gpu_burn
          resources:
            limits:
            nvidia.com/gpu: 2 # 申请 MIG GPU 的数量
          imagePullPolicy: Always
      restartPolicy: Always
resources:
  limits:
  nvidia.com/gpu: 2
  1. MIG Mixed 模式:
apiVersion: apps/v1
kind: Deployment
metadata:
  name: mig-demo
  namespace: default
spec:
  replicas: 1
  selector:
    matchLabels:
      app: mig-demo
  template:
    metadata:
      creationTimestamp: null
      labels:
        app: mig-demo
    spec:
      containers:
        - name: mig-demo1
          image: chrstnhntschl/gpu_burn
          resources:
            limits:
            nvidia.com/mig-4g.20gb: 1 # 
          imagePullPolicy: Always
      restartPolicy: Always
resources:
  limits:
  nvidia.com/gpu: 2
resources:
  limits:
  nvidia.com/mig-4g.20gb: 1 # 通过 nvidia.com/mig-g.gb 的资源类型公开各个 MIG 设备。
  1. 进入容器后可以查看只使用了一个 MIG 设备。

    mig03

评论