系统学习 AI 学习路径:从 GPU 运行时到推理、微调与 RAG 评估

nvidia-mig

NVIDIA MIG 分区:把一张 GPU 切成多个隔离实例,按固定 profile 分配显存与算力

补充说明

MIG(Multi-Instance GPU)把支持该特性的数据中心 GPU 切分成多个独立实例,每个实例有固定的显存和计算切片,故障与内存错误在实例之间隔离。常见于 A100、A30、H100、H200 等型号,具体支持情况取决于 GPU 型号、驱动和 vGPU 配置;消费级卡通常不支持。MIG 适合多租户推理、小模型隔离和资源配额,不适合需要整卡 NVLink 带宽的大规模训练。

MIG 的 profile 以“计算切片数 + 显存”命名,例如 A100 40GB 上的 1g.5gb、2g.10gb、3g.20gb、7g.40gb;H100 80GB 的 profile 和可用组合不同。创建实例前先用 nvidia-smi mig -lgip 看当前卡支持的 profile,不要照搬其他型号的命名。

常用示例

# 查看 GPU 是否支持 MIG、当前 MIG 模式和实例
nvidia-smi -L
nvidia-smi mig -lgip

# 启用 MIG 模式(可能需要重启或确保 GPU 空闲)
sudo nvidia-smi -i 0 -mig 1

# 创建两个 1g.10gb 的 GPU 实例,并同时创建计算实例
sudo nvidia-smi mig -i 0 -cgi 1g.10gb,1g.10gb -C

# 查看创建结果
nvidia-smi -L
# 删除计算实例和 GPU 实例(先删计算实例)
sudo nvidia-smi mig -i 0 -dci
sudo nvidia-smi mig -i 0 -dgi

# 关闭 MIG,恢复整卡
sudo nvidia-smi -i 0 -mig 0

# 查看 MIG 设备的显存和利用率
nvidia-smi -L
nvidia-smi mig -i 0 -lgi

Kubernetes 侧通常由 NVIDIA GPU Operator 管理 MIG,资源名类似 nvidia.com/mig-1g.10gb;集群需要选择 single 或 mixed MIG 策略,节点上的 MIG 配置要与 device plugin 的标签一致。

易错点与排错

相关命令

参考链接

读完自测

合上页面,先回答,再展开对照。

  1. 1g.10gb 这类 MIG profile 中的两个数字分别表示什么?
  2. 启用 MIG 前为什么要先停止 GPU 上的任务?
  3. MIG 实例的带宽和整卡是简单按比例切分吗?
答案
  1. 前面的 1g 表示计算切片数量,后面的 10gb 表示该实例可用的显存容量。
  2. MIG 模式切换会重置 GPU 分区,正在运行的任务会中断;应先停止容器、训练和推理进程,必要时重启节点。
  3. 不是。显存和计算切片是隔离的,但带宽、缓存、NVLink 等资源并非线性切分;性能要按具体实例和负载实测。
微信小程序码:扫码打开命令速查小程序

小程序版

微信扫码,随时查命令