docs.pytorch.org··3 年前
PyTorch 中的 Channels Last 内存格式
Channels Last Memory Format in PyTorch
PyTorch 官方中级教程,介绍 channels last 内存格式是什么,以及如何用它提升某些算子的性能。正文说明该格式在保留 NCHW 维度顺序的前提下改变内存排列,使通道成为最密集的维度,并利用现有 strides 结构实现,例如 10x3x16x16 批次在该格式下 strides 为 (768, 1, 48, 3);该格式仅针对 4D NCHW 张量实现,页面还给出连续格式与 channels last 格式之间的张量转换 API。前置条件为 PyTorch v1.5.0 和支持 CUDA 的 GPU。
- 分级
- 值得看
- 所属主题
- 模型服务与算力
- 收录时间
- 2026-10-29
推荐阅读
用 Green Contexts 控制 GPU 如何共享工作
Control How Your GPU Shares Work with Green Contexts
NVIDIA 技术博客的一篇文章,介绍 CUDA 13.1 中的 green contexts:允许应用在单个进程内显式划分 SM 和工作队列等 GPU 执行资源。文中举例说,为 green context 分配专用 SM 后,延迟敏感的 kernel 无需等待批量任务排空,在拥有 148 个 SM 的 Blackwell GPU 上可将关键 kernel 延迟从 0.140 ms 降至 0.007 ms。编程模型从 green context 句柄创建 stream,而不再依赖隐式的线程本地设备状态,同时保留既有基于 stream 的 CUDA 工作流;green contexts 为可选、可叠加特性,现有应用可按需逐步采用。
解锁连续批处理中的异步性
Unlocking asynchronicity in continuous batching
Hugging Face 博客于 2026 年 5 月 14 日发布的文章,作者为 Rémi Ouazan Reboul、Pedro Cuenca 和 Aritra Roy Gosthipaty,是高效 LLM 推理系列的第二篇。文章说明如何分离 CPU 与 GPU 工作负载以大幅提升推理性能:默认的同步连续批处理会导致 CPU 与 GPU 相互等待,停顿可能占近四分之一运行时间,因此采用异步批处理让 GPU 持续计算。文中提及 KV 缓存、FlashAttention、注意力掩码等概念,并以 H200 在 Inference Endpoints 上的价格为例说明成本考量。
使用 NVIDIA Megatron Core 扩展按位确定性预训练
Scale Bitwise-Deterministic Pretraining with NVIDIA Megatron Core
NVIDIA 技术博客的一篇文章,介绍 Megatron-LM PR #7262 提出的工作流:记录各 rank 有序的张量指纹,用于定位并修复大规模预训练中的非确定性。文中说明按位确定性可支持故障重放、loss 尖峰调试、系统变更验证以及中断的万亿参数训练恢复;在三项 Nemotron 工作负载上优化后,在 2,432 块 GPU 上把确定性开销降至约 2%,并在 800 步内保持按位确定性。还提到通过为 grouped-GEMM 写入者分配私有输出槽位、采用固定归约顺序等内核级改动,在保持并行的同时恢复确定性结果;Megatron Core 通过配方验证、内核测试和跨并行配置(含 FP8、FP4)的模块验证来防止回退。
用一个 Hub Bucket 传输万亿参数:TRL 中的增量权重同步
Shipping a Trillion Parameters With a Hub Bucket: Delta Weight Sync in TRL
文章介绍 TRL 中一项名为 Delta Weight Sync 的改动:异步强化学习中训练器每步需向推理引擎发送整个模型,而相邻两个优化步之间约 99% 的 bf16 权重完全相同,因此只把变化元素编码为稀疏 safetensors 文件,上传到 Hugging Face Bucket 并让 vLLM 拉取。文中称在 Qwen3-0.6B 上每步传输量从 1.2 GB 降至 20–35 MB,并演示了训练器、vLLM 与 Wordle 环境分处不同位置的分离式训练。
斯坦福 CS336:从零构建语言模型(2026 春)
斯坦福大学 2026 年春季 CS336 “Language Modeling from Scratch” 课程主页,授课教师为 Tatsunori Hashimoto 和 Percy Liang,另列有多名课程助理及各自的答疑时间,课程录像发布在 YouTube 播放列表。页面介绍课程定位:语言模型是现代自然语言处理应用的基础,课程通过带学生走完开发语言模型的完整流程,帮助其全面理解语言模型。页面还提供往期开课(2025 春、2024 春)链接和课程联系方式说明。课程涉及大规模模型训练,适合具备深度学习与系统实现基础、希望深入理解语言模型构建过程的人了解。
完全分片数据并行(FSDP2)入门
Getting Started with Fully Sharded Data Parallel (FSDP2)
PyTorch 官方中级教程,介绍 FSDP2 的工作方式,示例代码来自 pytorch/examples。正文对比 DDP,说明 FSDP 通过分片模型参数、梯度和优化器状态来降低 GPU 显存占用,使单卡放不下的模型也能训练;并描述前向与反向计算之外参数处于完全分片状态、计算前对分片参数做 all-gather、反向中对未分片梯度做 reduce-scatter 等流程。页面说明 FSDP1 已弃用,其教程已归档。