Alauda Build of Hami
  • 简体中文

    • 分配特定的设备内存

    目录

    向容器分配特定的设备内存

    要分配一定大小的 GPU 设备内存，只需在 nvidia.com/gpu 之外分配 nvidia.com/gpumem。

    apiVersion: v1
kind: Pod
metadata:
  name: gpu-pod
spec:
  containers:
    - name: ubuntu-container
      image: ubuntu:18.04
      command: ["bash", "-c", "sleep 86400"]
      resources:
        limits:
          nvidia.com/gpu: 2 # 请求 2 个虚拟 GPU
          nvidia.com/gpumem: 3000 # 每个虚拟 GPU 请求 3G 设备内存

    注意： nvidia.com/gpumem 不能与 nvidia.com/gpumem-percentage 一起使用