分配设备内存

向容器分配设备内存

通过指定 nvidia.com/gpumem 资源来分配一定量的设备内存。此项为可选;nvidia.com/gpumem 的每个单位等于 1 MiB。

      resources:
        limits:
          nvidia.com/gpualloc: 1 # requesting 1 GPU
          nvidia.com/gpumem: 3000 # Each GPU contains 3000m device memory

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

      resources:
        limits:
          nvidia.com/gpualloc: 1 # requesting 1 GPU
          nvidia.com/gpumem-percentage: 50 # Each GPU contains 50% device memory

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