6.1 张量并行与流水线并行:把大模型摊到多卡上


文档摘要

6.1 张量并行与流水线并行:把大模型摊到多卡上 本节摘要:张量并行把一层内的权重矩阵切片,多卡各算一块、每步前向互相同步;流水线并行把模型的层分组接力,各卡负责一段。前者通信密集、适合单机 NVLink 互联,后者通信稀疏、适合跨机扩展。本节讲两种切法的原理、气泡与通信的代价模型,以及在 vLLM 里各对应的一行启动参数。 会员。《6.1 张量并行与流水线并行:把大模型摊到多卡上》收录于灏天文库文集《vLLM》,原作者/来源:灏天文库,整理自「灏天文库」,提供技术教程、实践指南与问题解决方案,支持在线阅读、全文检索与知识沉淀,助力开发者系统化学习。本站整理收录,版权归原作者/开源协议所有。

该文档为会员专享,请先登录或注册后再查看


作者与出处
原作者: 灏天文库
来源:灏天文库
整理: 灏天文库整理
由灏天文库平台收录,内容或由平台用户上传,仅供学习交流
发布者: 作者: 灏天文库 转发
评论区 (0)
U