大模型百万Token长上下文训练教程(非常详细),上下文并行与Ring Attention从入门到精通,收藏这一篇就够了!

news2026/3/18 10:33:42
只用了几年时间上下文窗口就从 4k 膨胀到 1000 万。Meta 发布的 Llama 4 Scout 的时候说这个模型支持 1000 万 Token是 Llama 3 那 128k 的 78 倍。而Google Gemini 3 Pro 是 100 万Claude 4 也桐乡市100万。一次推理跑完整个代码库、几百篇论文、连续好几天的对话记录在技术上可行了但问题是硬件跟不上。405B 参数的模型32 位精度下光权重就要 6.5TB 内存。再算上梯度、状态、激活值后者还随上下文长度二次方增长。单台 NVIDIA HGX B300 配了 2.3TB HBM3e都不够。这就逼着必须做多节点分布式训练和推理几十上百块 NVIDIA Blackwell GPU 、NVLink 再加上 InfiniBand就成了数据中心的标配。所以难点就变味了 GPU 之间的通信瓶颈。并行化基础模型或数据集超出单卡容量就得上并行策略但是每种策略本质上都是拿通信开销换内存空间。数据并行是最直接的方案整个模型复制到每张卡上训练数据切开每张卡跑不同的 batch跑完一步同步梯度。适合小模型计算是瓶颈、内存不是问题的场景。模型并行针对大模型单卡装不下就把模型拆开不同的层放不同的卡上按顺序跑。405B 这种规模只能这样并且下游的卡得等上游算完中间是有空转的。张量并行更极端连单个矩阵乘法都塞不进一张卡。就需要把矩阵按行或按列切开分到各卡上算再通过 all-reduce 合起来。但这些都有共同的局限。模型大、上下文又长到几百万 Token张量并行也顶不住。因为注意力的二次方内存增长太凶激活值直接占满显存。128k 上下文的激活值内存是 8k 的 16 倍这个目前没办法因为就是这么夸张。上下文并行与序列并行序列并行和上下文并行都是在设备间切序列来省内存但切法不一样。序列并行配合张量并行使用只切那些非矩阵乘法的操作比如层归一化、dropout。张量并行管不到的地方序列并行接手每张卡处理一部分激活值。两者配合能把序列撑长一些但到 128k 以上还是会有问题因为注意力的二次方增长是绕不过去。上下文并行更彻底整个序列在所有模块里都切开包括注意力。每个操作拿到的都是分区后的序列。百万级上下文的训练就靠这个把激活值的内存占用分摊到各卡上。注意力一直是最麻烦的问题因为模型的其他操作基本都是逐 Token 独立处理并行起来很自然。但注意力不行每个 Token 都要看序列里所有其他 Token。序列切到多张卡上之后GPU 1 的 Token 怎么看 GPU 2 的 Token直接等数据传完再算整个流水线就卡住了。Ring Attention 就是来解决这个问题的让多节点多卡的大模型训练和推理能在大规模数据中心里跑起来。Zig Zag Ring Attention通信和计算重叠Ring Attention 把 GPU 组织成环形拓扑。每张卡的工作流程是这样的持有序列中 Q、K、V 张量的一个分块用本地的 K 和 V 给自己的 Q 分块算注意力把 K 和 V 传给环里的下一张卡从上一张卡接收 K 和 V循环往复直到所有 Q Token 都跟所有 K/V Token 算完注意力。关键在于计算和通信是重叠的。GPU 1 拿着当前的 K/V 分块算注意力的时候同时在从 GPU 0 接收下一批分块。通信延迟减少了因为不用干等数据全到了再开算。GPT 这类自回归模型有个额外的麻烦Token 只能看前面的 Token不能看后面的。所以会导致负载不均衡有些卡会空转Zig-Zag Ring Attention 解决这个问题的办法是交错分配不是按顺序切块而是 GPU 0 拿 Token [0, 4, 8…]GPU 1 拿 [1, 5, 9…]以此类推。每张卡都拿到早期和晚期 Token 的混合因果注意力计算时负载就均衡了环里不会有卡闲着。但是代价是索引逻辑稍微复杂一点不过大规模场景下性能收益很可观因果掩码下也能做到接近满 GPU 利用率。上下文并行与 Ring Attention 常见问题上下文并行把输入序列切到多张 GPU 上突破训练时的内存限制。跟张量并行、数据并行不同它在所有模型模块里都切序列维度。单卡装不下的百万级 Token 上下文只有靠这个才能训。Ring Attention 把 GPU 排成环每张卡一边算当前数据的注意力一边把键值对往下传。通信和计算重叠全对全的注意力计算不用等完整序列数据到齐GPU 不会干等。而序列并行只切非矩阵乘法操作层归一化之类的配合张量并行用。上下文并行在所有模块里都切序列包括注意力。超过 128k Token 的上下文必须用后者因为激活值内存二次方增长太猛了。为什么 Zig-Zag Ring Attention 比标准 Ring Attention 更好Zig-Zag 用交错分配代替顺序分配因果掩码计算时各卡负载更均衡。标准 Ring Attention 会让后面的卡等前面的分块造成计算空闲。Zig-Zag 把早期和晚期 Token 均匀撒到各卡上避免这个问题。那么训练百万级 Token 上下文的模型需要什么硬件多节点 GPU 集群配 HBM 内存加高速互连——NVIDIA NVLink 1.8TB/s 或者 InfiniBand。405B 参数模型 32 位精度从头训练加推理4 台 NVIDIA HGX B300 的机架部署是个不错的起点。总结上下文并行本质上是拿通信开销换内存空间而网络带宽是最要命的瓶颈。Ring Attention 要在 GPU 之间不停交换键值对传输时间一旦超过计算时间各卡就会从边算边传退化成等数据。NVIDIA NVLink 1.8TB/s 加 InfiniBand 的高速互连在多机架部署里不是可选项是必需品。互连带宽必须匹配 GPU 计算吞吐量否则上下文并行的效果会大打折扣。学AI大模型的正确顺序千万不要搞错了2026年AI风口已来各行各业的AI渗透肉眼可见超多公司要么转型做AI相关产品要么高薪挖AI技术人才机遇直接摆在眼前有往AI方向发展或者本身有后端编程基础的朋友直接冲AI大模型应用开发转岗超合适就算暂时不打算转岗了解大模型、RAG、Prompt、Agent这些热门概念能上手做简单项目也绝对是求职加分王给大家整理了超全最新的AI大模型应用开发学习清单和资料手把手帮你快速入门学习路线:✅大模型基础认知—大模型核心原理、发展历程、主流模型GPT、文心一言等特点解析✅核心技术模块—RAG检索增强生成、Prompt工程实战、Agent智能体开发逻辑✅开发基础能力—Python进阶、API接口调用、大模型开发框架LangChain等实操✅应用场景开发—智能问答系统、企业知识库、AIGC内容生成工具、行业定制化大模型应用✅项目落地流程—需求拆解、技术选型、模型调优、测试上线、运维迭代✅面试求职冲刺—岗位JD解析、简历AI项目包装、高频面试题汇总、模拟面经以上6大模块看似清晰好上手实则每个部分都有扎实的核心内容需要吃透我把大模型的学习全流程已经整理好了抓住AI时代风口轻松解锁职业新可能希望大家都能把握机遇实现薪资/职业跃迁这份完整版的大模型 AI 学习资料已经上传CSDN朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费】

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2408552.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

SpringBoot-17-MyBatis动态SQL标签之常用标签

文章目录 1 代码1.1 实体User.java1.2 接口UserMapper.java1.3 映射UserMapper.xml1.3.1 标签if1.3.2 标签if和where1.3.3 标签choose和when和otherwise1.4 UserController.java2 常用动态SQL标签2.1 标签set2.1.1 UserMapper.java2.1.2 UserMapper.xml2.1.3 UserController.ja…

wordpress后台更新后 前端没变化的解决方法

使用siteground主机的wordpress网站,会出现更新了网站内容和修改了php模板文件、js文件、css文件、图片文件后,网站没有变化的情况。 不熟悉siteground主机的新手,遇到这个问题,就很抓狂,明明是哪都没操作错误&#x…

网络编程(Modbus进阶)

思维导图 Modbus RTU(先学一点理论) 概念 Modbus RTU 是工业自动化领域 最广泛应用的串行通信协议,由 Modicon 公司(现施耐德电气)于 1979 年推出。它以 高效率、强健性、易实现的特点成为工业控制系统的通信标准。 包…

UE5 学习系列(二)用户操作界面及介绍

这篇博客是 UE5 学习系列博客的第二篇,在第一篇的基础上展开这篇内容。博客参考的 B 站视频资料和第一篇的链接如下: 【Note】:如果你已经完成安装等操作,可以只执行第一篇博客中 2. 新建一个空白游戏项目 章节操作,重…

IDEA运行Tomcat出现乱码问题解决汇总

最近正值期末周,有很多同学在写期末Java web作业时,运行tomcat出现乱码问题,经过多次解决与研究,我做了如下整理: 原因: IDEA本身编码与tomcat的编码与Windows编码不同导致,Windows 系统控制台…

利用最小二乘法找圆心和半径

#include <iostream> #include <vector> #include <cmath> #include <Eigen/Dense> // 需安装Eigen库用于矩阵运算 // 定义点结构 struct Point { double x, y; Point(double x_, double y_) : x(x_), y(y_) {} }; // 最小二乘法求圆心和半径 …

使用docker在3台服务器上搭建基于redis 6.x的一主两从三台均是哨兵模式

一、环境及版本说明 如果服务器已经安装了docker,则忽略此步骤,如果没有安装,则可以按照一下方式安装: 1. 在线安装(有互联网环境): 请看我这篇文章 传送阵>> 点我查看 2. 离线安装(内网环境):请看我这篇文章 传送阵>> 点我查看 说明&#xff1a;假设每台服务器已…

XML Group端口详解

在XML数据映射过程中&#xff0c;经常需要对数据进行分组聚合操作。例如&#xff0c;当处理包含多个物料明细的XML文件时&#xff0c;可能需要将相同物料号的明细归为一组&#xff0c;或对相同物料号的数量进行求和计算。传统实现方式通常需要编写脚本代码&#xff0c;增加了开…

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器的上位机配置操作说明

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器专为工业环境精心打造&#xff0c;完美适配AGV和无人叉车。同时&#xff0c;集成以太网与语音合成技术&#xff0c;为各类高级系统&#xff08;如MES、调度系统、库位管理、立库等&#xff09;提供高效便捷的语音交互体验。 L…

(LeetCode 每日一题) 3442. 奇偶频次间的最大差值 I (哈希、字符串)

题目&#xff1a;3442. 奇偶频次间的最大差值 I 思路 &#xff1a;哈希&#xff0c;时间复杂度0(n)。 用哈希表来记录每个字符串中字符的分布情况&#xff0c;哈希表这里用数组即可实现。 C版本&#xff1a; class Solution { public:int maxDifference(string s) {int a[26]…

【大模型RAG】拍照搜题技术架构速览:三层管道、两级检索、兜底大模型

摘要 拍照搜题系统采用“三层管道&#xff08;多模态 OCR → 语义检索 → 答案渲染&#xff09;、两级检索&#xff08;倒排 BM25 向量 HNSW&#xff09;并以大语言模型兜底”的整体框架&#xff1a; 多模态 OCR 层 将题目图片经过超分、去噪、倾斜校正后&#xff0c;分别用…

【Axure高保真原型】引导弹窗

今天和大家中分享引导弹窗的原型模板&#xff0c;载入页面后&#xff0c;会显示引导弹窗&#xff0c;适用于引导用户使用页面&#xff0c;点击完成后&#xff0c;会显示下一个引导弹窗&#xff0c;直至最后一个引导弹窗完成后进入首页。具体效果可以点击下方视频观看或打开下方…

接口测试中缓存处理策略

在接口测试中&#xff0c;缓存处理策略是一个关键环节&#xff0c;直接影响测试结果的准确性和可靠性。合理的缓存处理策略能够确保测试环境的一致性&#xff0c;避免因缓存数据导致的测试偏差。以下是接口测试中常见的缓存处理策略及其详细说明&#xff1a; 一、缓存处理的核…

龙虎榜——20250610

上证指数放量收阴线&#xff0c;个股多数下跌&#xff0c;盘中受消息影响大幅波动。 深证指数放量收阴线形成顶分型&#xff0c;指数短线有调整的需求&#xff0c;大概需要一两天。 2025年6月10日龙虎榜行业方向分析 1. 金融科技 代表标的&#xff1a;御银股份、雄帝科技 驱动…

观成科技:隐蔽隧道工具Ligolo-ng加密流量分析

1.工具介绍 Ligolo-ng是一款由go编写的高效隧道工具&#xff0c;该工具基于TUN接口实现其功能&#xff0c;利用反向TCP/TLS连接建立一条隐蔽的通信信道&#xff0c;支持使用Let’s Encrypt自动生成证书。Ligolo-ng的通信隐蔽性体现在其支持多种连接方式&#xff0c;适应复杂网…

铭豹扩展坞 USB转网口 突然无法识别解决方法

当 USB 转网口扩展坞在一台笔记本上无法识别,但在其他电脑上正常工作时,问题通常出在笔记本自身或其与扩展坞的兼容性上。以下是系统化的定位思路和排查步骤,帮助你快速找到故障原因: 背景: 一个M-pard(铭豹)扩展坞的网卡突然无法识别了,扩展出来的三个USB接口正常。…

未来机器人的大脑:如何用神经网络模拟器实现更智能的决策?

编辑&#xff1a;陈萍萍的公主一点人工一点智能 未来机器人的大脑&#xff1a;如何用神经网络模拟器实现更智能的决策&#xff1f;RWM通过双自回归机制有效解决了复合误差、部分可观测性和随机动力学等关键挑战&#xff0c;在不依赖领域特定归纳偏见的条件下实现了卓越的预测准…

Linux应用开发之网络套接字编程(实例篇)

服务端与客户端单连接 服务端代码 #include <sys/socket.h> #include <sys/types.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <arpa/inet.h> #include <pthread.h> …

华为云AI开发平台ModelArts

华为云ModelArts&#xff1a;重塑AI开发流程的“智能引擎”与“创新加速器”&#xff01; 在人工智能浪潮席卷全球的2025年&#xff0c;企业拥抱AI的意愿空前高涨&#xff0c;但技术门槛高、流程复杂、资源投入巨大的现实&#xff0c;却让许多创新构想止步于实验室。数据科学家…

深度学习在微纳光子学中的应用

深度学习在微纳光子学中的主要应用方向 深度学习与微纳光子学的结合主要集中在以下几个方向&#xff1a; 逆向设计 通过神经网络快速预测微纳结构的光学响应&#xff0c;替代传统耗时的数值模拟方法。例如设计超表面、光子晶体等结构。 特征提取与优化 从复杂的光学数据中自…