分配设备内存

目录

向容器分配设备内存

通过指定资源如 nvidia.com/gpumem 分配一定大小的设备内存。可选,每个单位的 nvidia.com/gpumem 等于 1M。

      resources:
        limits:
          nvidia.com/gpu: 1 # 请求 1 个 GPU
          nvidia.com/gpumem: 3000 # 每个 GPU 包含 3000m 设备内存

通过指定资源 nvidia.com/gpumem-percentage 分配设备内存的百分比。可选,每个单位的 nvidia.com/gpumem-percentage 等于设备内存的 1%。

      resources:
        limits:
          nvidia.com/gpu: 1 # 请求 1 个 GPU
          nvidia.com/gpumem-percentage: 50 # 每个 GPU 包含 50% 设备内存

注意: nvidia.com/gpumemnvidia.com/gpumem-percentage 不能同时分配