监控容器服务中的 AI 训练任务

在 AI 训练模型处理业务用例时，监控训练任务状态和底层资源负载至关重要。本文为您介绍在容器服务集群中监控 AI 训练任务的方法。

前提条件

• 集群已接入托管 Prometheus，并同步安装了 node-exporter 组件。详情请参见 接入托管 Prometheus。

• 已安装 rdma-device-plugin 组件，详情请参见 安装组件。
• 已安装 nvidia-device-plugin 组件，并同步安装了 dcgm-exporter。详情请参见 安装组件。
• 已创建告警通知策略，详情请参见 创建告警通知策略。
• 已创建告警聚合策略，详情请参见 创建告警聚合策略。

操作步骤

步骤一：配置 AI 训练任务

您可以在集群中使用工作负载部署 AI 训练任务，并使用 GPU 和 RDMA 资源。配置方法请参见 NVIDIA GPU 调度 和 VKE 集群中使用 RDMA 资源。

为了能够在大盘中查看到训练任务的相关信息，您需要在训练任务中增加ai.vke.volcengine.com/task、ai.vke.volcengine.com/job、ai.vke.volcengine.com/team和ai.vke.volcengine.com/user标签，示例如下。

apiVersion: apps/v1
kind: Deployment
metadata:
  name: ai-demo # 训练任务的名称
  namespace: default # 训练任务所在的命名空间
spec: 
  replicas: 1 # 训练任务的副本数
  selector: 
    matchLabels: 
      app: ai-demo
  template: 
    metadata: 
      labels: 
        app: ai-demo
        ai.vke.volcengine.com/task: master # 训练任务的角色，例如 master、chief、worker
        ai.vke.volcengine.com/job: ai-name # 训练任务的名称
        ai.vke.volcengine.com/team: team-a # 训练任务的归属团队
        ai.vke.volcengine.com/user: user-a # 训练任务的归属成员
    spec:
      containers:
      - name: ai # 容器名称
        image: doc-cn-beijing.cr.volces.com/vke/gpu-burn:1.0 # 训练任务的镜像和版本
        command: [ "/bin/bash", "-c", "--" ]
        args: [ "while true; do sleep 3600; done;" ]
        resources:
          limits:
            nvidia.com/gpu: "1" # 按需配置 GPU 显卡数量

步骤二：搭建 Grafana 并导入监控大盘

当需要对集群中运行的 AI 训练任务以及底层资源进行监控时，您可以下载本文中提供的监控大盘，并导入到 Grafana 系统中使用。详情请参见 AI 常用监控大盘。

操作步骤如下：

1. 在 VKE 集群中部署 Grafana 并接入工作区。详情请参见 在容器服务集群中部署 Grafana 并接入工作区。
2. 在本地环境中使用 kubectl 连接集群，并执行以下命令，配置 Grafana 服务的端口映射。

kubectl port-forward service/grafana 3000:3000 -n volcano-metrics

3. 在本地环境中打开浏览器，在地址栏中输入 localhost:3000，进入 Grafana 登录界面，使用默认用户名admin和密码admin登录。
4. 在左侧导航栏中，选择 Dashboards，进入监控大盘配置页面。
   
5. 单击 New ，在下拉菜单中选择 import，进入导入大盘页面。
   
6. 单击 Upload JSON file，并选择保存在本地的监控大盘 JSON 文件。大盘 JSON 文件，请参见 监控大盘说明。
   
7. 配置监控大盘的基本信息。
   

   配置项	说明
   Name	（可选）根据需要修改监控大盘的名称。
   Folder	（可选）选择大盘保存的文件夹，不修改则默认保存在 General 文件夹。
   VMP	选择数据源，您需要在下拉菜单中选择与目标集群绑定的托管 Prometheus 工作区。

8. 单击 import，完成导入，即可查看监控大盘。
   

步骤三：开启 AI 资源观测

1. 登录 容器服务控制台。
2. 单击左侧导航栏中的 集群。
3. 在集群列表页面，单击需要查看事件的目标集群。
4. 在集群管理页面的左侧导航栏中，单击 观测配置。
5. 选择 AI 资源 卡片，单击 启用，开启 AI 资源观测。
   
6. 系统自动检查开启集群观测所需的必要条件。包括：工作区配置、组件状态等。
   
7. 单击 开启，开启 AI 资源观测。

步骤四：配置告警

1. 在集群管理页面的左侧导航栏中，单击 观测配置。

2. 选择 AI 资源 卡片，单击 编辑配置 并选择 告警 页签，配置告警的相关参数。
   

   配置项	说明
   告警模版	勾选需要的告警模版，允许多选。
   告警聚合策略	在下拉菜单中选择告警聚合策略。详情请参见 创建告警聚合策略。
   告警通知策略	在下拉菜单中选择告警通知策略。系统会使用通知策略中配置的告警等级和联系人组，将告警发送给指定的联系人。详情请参见 创建告警通知策略。

3. 单击 确定，完成配置。

结果验证

查看监控大盘

1. 在本地环境中使用 kubectl 连接集群，并执行以下命令，配置 Grafana 服务的端口映射。

kubectl port-forward service/grafana 3000:3000 -n volcano-metrics

2. 在本地环境中打开浏览器，在地址栏中输入 localhost:3000，进入 Grafana 登录界面，使用默认用户名admin和密码admin登录。
3. 在左侧导航栏中，选择 Dashboards，即可查看 AI 训练任务的监控大盘。
   

查看告警配置

1. 登录 VMP 服务控制台。
2. 单击左侧导航栏的 告警中心 > 告警规则，选择 告警规则组 页签。在告警规则组列表中，查看生成的告警规则组。
   
3. 单击告警规则组名称，进入告警规则组详情页面。选择 告警规则列表 页签，查看该告警规则组下的所有告警规则。您可以根据需要启用、关闭、编辑对应的告警规则。具体操作方式，请参见 管理告警规则 和 管理告警规则组。
   

AI 常用监控大盘

集群 AI 监控大盘

该大盘主要展示了 集群维度 的 AI 训练任务信息和资源使用信息。

集群 AI 监控大盘-20230912.json

55.74KB

节点 GPU 大盘

该大盘主要展示了 节点维度 的监控信息，包括 GPU 节点监控信息和 GPU 卡监控信息。

节点 GPU 大盘-20230912.json

49.49KB

训练任务大盘(rdma-device-plugin)

该大盘展示了 任务维度 的 AI 训练任务监控信息和基础资源（GPU、RDMA）监控信息。

训练任务大盘(rdma-device-plugin)-20230920.json

91.68KB

训练任务大盘(node-exporter)

该大盘展示了 任务维度 的 AI 训练任务监控信息和基础资源（GPU、RDMA）监控信息。

训练任务大盘(node-exporter)-20230912.json

74.17KB