首页/AI/大模型部署与运维速查/张量并行:单卡装不下的标准解法
本包目录(267 条)
部署形态与选型43 条推理环境与 GPU50 条推理参数与优化50 条
显存与并发估算40 条
服务化与网关34 条运维监控与排障50 条

张量并行:单卡装不下的标准解法

把每层权重与计算切到多卡:每卡只需驻留约 1/TP 的权重与对应 KV;vLLM 用 --tensor-parallel-size 指定,是「单卡装不下」场景的常规扩展方式。

启动示例

启动示例
vllm serve <模型路径> --tensor-parallel-size 4   # 4 卡张量并行
TP 每层都要通信:优先在 NVLink 互联的卡组内做;跨节点 TP 要评估网络带宽与延迟。

易错点

  • 按卡数随意加大 TP:通信开销上升、效率打折,目标应是「够用且效率可接受」

数据来源与参考

  • vLLM 官方文档:并行与分布式(docs.vllm.ai)
  • Megatron-LM 论文:张量并行(arxiv.org/abs/1909.08053)
​‌‌​​‌​​​‌‌​‌‌‌‌​‌‌​​​‌‌​‌‌‌​​‌‌​​‌​‌‌‌​​‌‌‌​​​‌​‌‌​‌​​‌​‌‌‌​​​‌​‌‌​‌​​‌​‌‌​​​‌‌​‌‌‌​‌​​​‌‌​‌‌‌‌​​‌​‌‌‌​​‌‌​​​‌‌​‌‌​‌‌‌‌​‌‌​‌‌​‌