GPU-Manager 说明
最近更新时间: 2024-10-17 17:10:00
简介
组件介绍
GPU Manager 提供一个 All-in-One 的 GPU 管理器,基于 Kubernetes DevicePlugin 插件系统实现,该管理器提供了分配并共享 GPU、GPU 指标查询、容器运行前的 GPU 相关设备准备等功能,支持用户在 Kubernetes 集群中使用 GPU 设备。
组件功能
拓扑分配:提供基于 GPU 拓扑分配功能,当用户分配超过1张 GPU 卡的应用,可以选择拓扑连接最快的方式分配 GPU 设备。
GPU 共享:允许用户提交小于1张卡资源的任务,并提供 QoS 保证。
应用 GPU 指标的查询:用户可以访问主机端口(默认为 5678)的
/metric
路径,可以为 Prometheus 提供 GPU 指标的收集功能,访问/usage
路径可以进行可读性的容器状况查询。
部署在集群内的 Kubernetes 对象
Kubernetes 对象名称 | 类型 | 建议预留资源 | 所属 Namespaces |
---|---|---|---|
gpu-manager-daemonset | DaemonSet | 每节点1核 CPU, 1Gi内存 | kube-system |
gpu-quota-admission | Deployment | 每节点1核 CPU, 1Gi内存 | kube-system |
使用场景
在 Kubernetes 集群中运行 GPU 应用时,可以解决 AI 训练等场景中申请独立卡造成资源浪费的情况,让计算资源得到充分利用。
限制条件
该组件基于 Kubernetes DevicePlugin 实现,可直接在 Kubernetes 1.10 以上版本的集群使用。
每张 GPU 卡一共有100个单位的资源,仅支持0 - 1的小数卡,以及1的倍数的整数卡设置。显存资源是以256MiB为最小的一个单位的分配显存。
使用 GPU-Manager 要求集群内包含 GPU 机型节点。
使用方法
组件安装
登录容器服务控制台,在左侧导航栏中选择集群。
在“集群管理”页面单击目标集群 ID,进入集群详情页。
选择左侧菜单栏中的组件管理,进入 “组件列表” 页面。
在“组件列表”页面中选择新建,并在“新建组件”页面中勾选 GpuManager。
单击完成即可创建组件。
创建细粒度的 GPU 工作负载
当 GpuManager 组件成功安装后,您可通过以下两种方式创建细粒度的 GPU 工作负载。
方式一:通过 TKE 控制台创建
登录容器服务控制台,选择左侧导航栏中的集群。
选择需要创建 GPU 应用的集群,进入工作负载管理页,并单击新建。
在“新建Workload”页面根据实际需求进行配置,可在“GPU资源”配置细粒度的 GPU 工作负载。
方式二:通过 yaml 创建
说明:在提交时通过 yaml 为容器设置 GPU 的使用资源,核资源需要在 resource 上填写 fincloud.tencent.cn/vcuda-core
,显存资源需要在 resource 上填写fincloud.tencent.cn/vcuda-memory
。
下面给出 yaml 示例:
- 使用1张卡的 P4 设备:
apiVersion: v1
kind: Pod
...
spec:
containers:
name:gpu
resources:fincloud.tencent.cn/vcuda-core:100
- 使用0.3张卡,5GiB 显存的应用:
apiVersion: v1
kind: Pod
...
spec:
containers:
name:gpu
resources:fincloud.tencent.cn/vcuda-core:30 fincloud.tencent.cn/vcuda-memory:20