- 文档首页
容器服务
容器服务 VKE
用户指南
异构计算
NVIDIA GPU
配置 NVIDIA GPU 调度
NVIDIA GPU
配置 NVIDIA GPU 调度
配置 NVIDIA GPU 调度
NVIDIA GPU 调度为容器服务中的基础 GPU 调度方式,允许容器占用一块或多块 GPU 整卡,并使用这些 GPU 卡的所有算力和显存。本文主要介绍在集群中使用 NVIDIA GPU 调度的配置方法。
已安装 nvidia-device-plugin 组件。详情请参见 安装组件。 步骤一:创建 GPU 节点池
- 在左侧导航栏中选择 集群,单击目标集群名称,进入集群控制台。
- 在节点池配置页面中,单击 创建节点池,根据您的实际需求,创建节点池。详细参数说明,请参见 创建节点池。其中:
- 在 计算规格 中,选择 GPU 计算型 规格,为集群添加包含 GPU 的节点。规格详细介绍,请参见 GPU 计算型。
- (可选)在 高级配置 中,单击 添加节点标签,为节点池添加 GPU 调度标签。本例中配置标签键为vke.node.gpu.schedule,标签值为nvidia。表示在该节点池中的节点上,开启 NVIDIA GPU 调度能力。
本文以创建无状态负载(Deployment)为例进行介绍,详细创建步骤参见 创建无状态负载。 - 在集群管理页面的左侧导航栏中,选择 工作负载 > 无状态负载。
- 创建无状态负载或更新已有的负载。其中,在 容器配置 步骤的 资源配额 处,根据需求配置 GPU 资源。
说明
根据工作负载所在的集群是否安装 mgpu 组件,控制台有如下不同显示:
- 未安装 mgpu 组件:系统显示 GPU 算力 和 显卡类型,请按实际需求配置参数即可。
- 已安装 mgpu 组件:系统显示 GPU 算力 和 GPU 显存 配置项。在 NVIDIA GPU 调度场景中,GPU 算力 必须输入大等于 1 的整数,此时 GPU 显存 会变为 显卡类型,且系统会显示 启用 nvidia 调度 按钮。
- NVIDIA GPU 调度场景,需要勾选 启用 nvidia 调度 按钮;若不勾选,表示不使用 NVIDIA GPU 调度,转为使用 mGPU 调度。mGPU 调度和 mgpu 组件目前是均为公测功能。
通过 YAML 创建工作负载,并实现 NVIDIA GPU 调度示例,如下所示:
执行以下命令,登录容器查看容器被分配显存总量。
预期结果如下,可以看到 Pod 使用的 GPU 卡类型、驱动等信息。
使用 NVIDIA GPU 调度后,支持通过云原生观测中的 AI 资源观测,查看 GPU 资源看板。详情请参见 AI 资源观测。
最近更新时间:2026.03.19 12:11:11