- 综合排序
- 最热优先
- 最新优先
- 来自专栏罗西的思考
模型并行分布式训练 Megatron (3) ---模型并行实现
[源码解析] 模型并行分布式训练 Megatron (3) ---模型并行实现 目录 [源码解析] 模型并行分布式训练 Megatron (3) ---模型并行实现 0x00 摘要 0x01 并行Transformer 6.2 共轭函数 0xFF 参考 0x00 摘要 NVIDIA Megatron 是一个基于 PyTorch 的分布式训练框架，用来训练超大Transformer语言模型，其通过综合应用了数据并行，Tensor 并行和Pipeline并行来复现 GPT3，值得我们深入分析其背后机理。 本文将看看 Megatron 如何处理模型并行。 本系列其他文章为： [源码解析] 模型并行分布式训练Megatron (1) --- 论文 & 基础 [源码解析] 模型并行分布式训练Megatron (2) --- 整体架构 0x01 并行Transformer
2.9K20编辑于 2022-11-28 - 来自专栏罗西的思考
模型并行分布式训练 Megatron (4) --- 如何设置各种并行
[源码解析] 模型并行分布式训练 Megatron (4) --- 如何设置各种并行 目录 [源码解析] 模型并行分布式训练 Megatron (4) --- 如何设置各种并行 0x00 摘要 0x01 Data-parallel 7.1 分组 7.2 如何使用 0x08 模型组 0x09 如何把模型分到GPU 0xFF 参考 0x00 摘要 NVIDIA Megatron 是一个基于 PyTorch 的分布式训练框架 本系列其他文章为： [源码解析] 模型并行分布式训练Megatron (1) --- 论文 & 基础 [源码解析] 模型并行分布式训练Megatron (2) --- 整体架构 [源码解析] 模型并行分布式训练 数据并行，模型并行，流水线并行这几种并行之中，每个节点分别属于哪个部分？ 如何避免流水线带来的问题。 我们接下来就仔细分析一下。 2.1 全局变量 因为前文_initialize_distributed之中调用了torch.distributed.init_process_group 初始化分布式环境，所以我们知道，每个进程都有自己的
3K10编辑于 2022-05-09 - 来自专栏机器之心
分布式训练中数据并行远远不够，「模型并行+数据并行」才是王道
来自加州大学洛杉矶分校和英伟达的研究人员探索了混合并行化方法，即结合数据并行化和模型并行化，解决 DP 的缺陷，实现更好的加速。 数据并行化（Data parallelism，DP）是应用最为广泛的并行策略，但随着数据并行训练设备数量的增加，设备之间的通信开销也在增长。 图 2：不同的训练并行化策略，2(a) 展示了数据并行化训练，2(b) 展示了模型并行化训练。 该研究发现，在规模较大的情况下，混合训练在最小化端到端训练时间方面比仅使用 DP 更加高效。 该研究的贡献如下： 当 DP 愈加低效时，可以使用混合并行化策略（即每个数据并行化 worker 在多个设备上也是模型并行化的）进一步扩展多设备训练。 这部分量化了使用数据并行策略的训练时间、使用模型并行策略的训练时间，以及使用混合并行策略的训练时间（详情参见原论文），并得出结论：在一定条件下，混合并行策略的效果优于仅使用 DP。
1.4K20发布于 2019-09-03 - 来自专栏CreateAMind
分布式并行架构Ray介绍
Ray is a fast and simple framework for building and running distributed applications. The same code can be run on a single machine to achieve efficient multiprocessing, and it can be used on a cluster for large computations.
2.1K20发布于 2019-08-13 - 来自专栏Michael阿明学习之路
python 并发、并行处理、分布式处理
并行编程 线程 进程 使用多个进程 接口 Executor ，ProcessPoolExecutor 5. 锁 6. 分布式处理 dask pyspark mpi4py 科学计算 7. "https://www.csdn.net"])) # 不会并行获取 "https://www.csdn.net"])) # 会并行但受制于 并行编程 问题是独立的，或者高度独立的，可以使用多核进行计算 如果子问题之间需要共享数据，实现起来不那么容器，有进程间通信开销的问题 线程 以共享内存方式实现并行的一种常见方式是 线程 由于 python 分布式处理 dask https://www.dask.org/ pyspark 用户提交任务，集群管理器自动将任务分派给空闲的执行器 mpi4py 科学计算 https://pypi.org/project
2.3K20编辑于 2022-09-22 - 来自专栏罗西的思考
PyTorch 分布式(18) --- 使用 RPC 的分布式管道并行
[源码解析] PyTorch 分布式(18) --- 使用 RPC 的分布式管道并行 目录 [源码解析] PyTorch 分布式(18) --- 使用 RPC 的分布式管道并行 0x00 摘要 0x01 6) ---- 引擎(下) [源码解析] PyTorch分布式优化器(1)----基石篇 [源码解析] PyTorch分布式优化器(2)----数据并行优化器 [源码解析] PyTorch分布式优化器( 3)---- 模型并行 [源码解析] PyTorch 分布式(14) --使用 Distributed Autograd 和 Distributed Optimizer [源码解析] PyTorch 分布式 这可以看作是单机模型并行最佳实践中讨论的多 GPU 流水线并行的分布式对应版本。 1.2 基础知识 之前的教程分布式 RPC 框架入门 展示了如何使用torch.distributed.rpc 为 RNN 模型实现分布式模型并行。
99940编辑于 2021-12-21 - 来自专栏罗西的思考
PyTorch分布式优化器(3)---- 模型并行
[源码解析] PyTorch分布式优化器(3)---- 模型并行 目录 [源码解析] PyTorch分布式优化器(3)---- 模型并行 0x00 摘要 0x01 前文回顾 0x02 单机模型 2.1 本文介绍PyTorch 分布式优化器和PipeDream之中的优化器，主要涉及模型并行（流水线并行）。 于是人们引入了模型并行（model parallel）。 与此对应，优化器也需要做不同的修改以适应模型并行的需求。为了更好的分析，本文首先介绍单机模型并行，然后介绍PyTorch分布式优化器。 模型并行被广泛用于分布式训练。 0x03 分布式问题和方案 我们已经了解了单机之上的模型并行，接下来就要看模型跨越多个服务器的分布式模型并行训练。 3.1 思路 我们先设想一下如果自己实现分布式优化器，应该如何处理。
1.8K40编辑于 2021-12-10 - 来自专栏罗西的思考
并行分布式框架 Celery 之架构 (1)
[源码解析] 并行分布式框架 Celery 之架构 (1) 目录 [源码解析] 并行分布式框架 Celery 之架构 (1) 0x00 摘要 0x01 Celery 简介 1.1 什么是 Celery 它是一个简单、灵活且可靠的，处理大量消息的分布式系统，专注于实时处理的异步任务队列，同时也支持任务调度。 0x02 Celery的架构 Celery 的基本逻辑为：分布式异步消息任务队列。 在 Celery 中，采用的是分布式的管理方式，每个节点之间都是通过广播/单播进行通信，从而达到协同效果。 3.2 Celery 辅助功能 以上为基础功能，但是作为分布式异步消息任务队列，我们还需要辅助功能（以及相关问题），比如。 以上功能哪些属于带有分布式特点的？ 3.3 如何划分 进一步问题是：这些辅助功能是作为基础功能模块的一部分？还是独立出来成为一个功能模块？
1.1K20编辑于 2022-05-09 - 来自专栏AI前沿技术
大模型分布式训练 —— 张量并行TP
本文：1）首先介绍张量并行矩阵分块，按行和按列两种方式，2）接着图解MLP和self-attention的分块原理，以及对应的通信量，3）最后介绍编码层和损失层的分块和通信优化。 3，自注意力层分块计算 参数矩阵分解方式：对三个参数矩阵Q，K，V，按照”列切割”，每个头放到一块GPU上，做并行计算。
30310编辑于 2026-01-13 - 来自专栏罗西的思考
并行分布式框架 Celery 之 容错机制
0x00 摘要 Celery是一个简单、灵活且可靠的，处理大量消息的分布式系统，专注于实时处理的异步任务队列，同时也支持任务调度。本文介绍 Celery 的故障转移容错机制。 这里宣传一个同学的分布式函数调度框架，https://github.com/ydf0509/distributed_framework。非常优秀的实现。大家可以作为 Celery 的替代品。
1.1K20发布于 2021-05-19 - 来自专栏数据结构和算法
探索LightGBM：并行化与分布式训练
导言 LightGBM是一种高效的梯度提升决策树算法，通过并行化和分布式训练，可以加速模型训练过程，特别是在处理大规模数据集时。 本教程将详细介绍如何在Python中使用LightGBM进行并行化和分布式训练，并提供相应的代码示例。 并行化训练 LightGBM支持多线程和多进程的并行化训练，可以利用多核CPU来加速模型训练。 除了并行化训练，LightGBM还支持分布式训练，可以在多台机器上同时训练模型。 我们介绍了如何利用多线程进行并行化训练，并演示了如何在多台机器上进行分布式训练。 通过这篇博客教程，您可以详细了解如何在Python中使用LightGBM进行并行化和分布式训练。 您可以根据需要对代码进行修改和扩展，以满足特定的并行化和分布式训练需求。
84010编辑于 2024-02-28 - 来自专栏罗西的思考
并行分布式框架 Celery 之架构 (2)
[源码解析] 并行分布式框架 Celery 之架构 (2) 0x00 摘要 Celery是一个简单、灵活且可靠的，处理大量消息的分布式系统，专注于实时处理的异步任务队列，同时也支持任务调度。 上文 [源码解析] 并行分布式框架 Celery 之架构 (1) 中，我们大致介绍了 Celery 的概念，用途和架构，现在回忆 Celery 的架构图如下： +-----------+ 这就涉及到两个最重要的问题： 如何拆分计算逻辑； 如何分发计算逻辑； 于是 Spark 把所有的计算逻辑划分为这两种类型： 能够分发到各个节点上并行执行的； 需要经过一定量的结果合并之后才能继续执行的； 有些人可能会担心，分块处理会导致并行性能下降，实际上，由于避免了消息传递的开销，因此反而会大大的提高性能。 我们从负载均衡，容灾恢复，worke之间交互这三个角度来看看 Celery 如何实现分布式。
1.2K10发布于 2021-04-01 - 来自专栏罗西的思考
模型并行分布式训练Megatron (1) --- 论文 & 基础
[源码解析] 模型并行分布式训练Megatron (1) --- 论文 & 基础 目录 [源码解析] 模型并行分布式训练Megatron (1) --- 论文 & 基础 0x00 摘要 0x01 Introduction 1.5 指导原则 Megatron 开发者研究了各种组合之间如何影响吞吐量，基于这些研究得出来分布式训练的一些指导原则： 不同的并行模式以复杂的方式互相作用：并行化策略影响通信量、核的计算效率，以及worker 分布式训练是通信密集型的。使用较慢的节点间连接或更多的通信密集型分区会阻碍性能。 FlexFlow是一个进行这种并行计算的深度学习框架，并且提供了一种选择最佳并行化策略的方法。最近，Mesh TensorFlow引入了一种语言，用于指定TensorFlow中的一般分布式张量计算。 （一） NVIDIA Megatron：超大Transformer语言模型的分布式训练框架 (二)
3.9K10编辑于 2022-05-09 - 来自专栏罗西的思考
并行分布式框架 Celery 之 worker 启动 (2)
[源码解析] 并行分布式框架 Celery 之 worker 启动 (2) 目录 [源码解析] 并行分布式框架 Celery 之 worker 启动 (2) 0x00 摘要 0x01 前文回顾 0x2 逐次调用step 3.2.4.1 Hub 3.2.4.2 Pool 3.2.4.3 Consumer 0x4 本文总结 0xFF 参考 0x00 摘要 Celery是一个简单、灵活且可靠的，处理大量消息的分布式系统 Kombu 之 Consumer [源码分析] 消息队列 Kombu 之 Producer [源码分析] 消息队列 Kombu 之 启动过程 [源码解析] 消息队列 Kombu 之 基本架构 以及 源码解析 并行分布式框架 Celery 之架构 (2) [源码解析] 并行分布式框架 Celery 之架构 (2) [源码解析] 并行分布式框架 Celery 之 worker 启动 (1) 0x01 前文回顾 前文提到了，我们经过一系列过程
1.6K20发布于 2021-04-07 - 来自专栏罗西的思考
并行分布式任务队列 Celery 之 负载均衡
[源码解析] 并行分布式任务队列 Celery 之 负载均衡 目录 [源码解析] 并行分布式任务队列 Celery 之 负载均衡 0x00 摘要 0x01 负载均衡 1.1 哪几个 queue 1.1.1 Autoscaler 2.1 调用时机 2.2 具体实现 2.2.1 bgThread 2.2.2 定义 0xEE 个人信息 0xFF 参考 0x00 摘要 Celery是一个简单、灵活且可靠的，处理大量消息的分布式系统
1K20发布于 2021-05-17 - 来自专栏罗西的思考
模型并行分布式训练Megatron (2) --- 整体架构
[源码解析] 模型并行分布式训练Megatron (2) --- 整体架构 目录 [源码解析] 模型并行分布式训练Megatron (2) --- 整体架构 0x00 摘要 0x01 启动 1.1 分布式启动 ，用来训练超大Transformer语言模型，其通过综合应用了数据并行，Tensor并行和Pipeline并行来复现 GPT3，值得我们深入分析其背后机理。 本系列其他文章为: [源码解析] 模型并行分布式训练Megatron (1) --- 论文 & 基础 0x01 启动 1.1 分布式启动 启动脚本在 examples/pretrain_bert_distributed.sh 注意，数据并行是把不同数据加载到不同的rank之上，而 Tensor模型并行组之中每个rank都加载同样数据。 调用 mpu.initialize_model_parallel 来设置模型并行，数据并行等各种进程组，我们下文会重点讨论。
3.4K20编辑于 2022-05-09 - 来自专栏Michael阿明学习之路
使用云函数进行分布式并行计算
云函数是一段运行在云端的代码，无需管理服务器，在开发工具内编写、一键上传部署即可运行后端代码。
74010编辑于 2023-12-08 - 来自专栏罗西的思考
并行分布式框架 Celery 之 worker 启动 (1)
0x00 摘要 Celery是一个简单、灵活且可靠的，处理大量消息的分布式系统，专注于实时处理的异步任务队列，同时也支持任务调度。Celery 是调用其Worker 组件来完成具体任务处理。 接下来，这个worker具有两套数据结构，这两套数据结构是并行运行的，他们分别是 'ET时刻表' 、就绪队列。
1.3K20发布于 2021-04-01 - 来自专栏AI研习社
问答 | tensorflow 怎么配置非数据并行分布式？
话不多说，直接上题 @iresine 问：tensorflow 怎么配置非数据并行分布式？ 来自社友的回答 ▼▼▼ @王奇文： 你指的是in-graph模式的模型并行吧？思考以下几个问题： ①cpu和gpu的优势和劣势分别是什么？ cpu擅长指定操作，逻辑判断，gpu擅长矩阵运算，二者相互补充，一边用cpu执行参数更新、逻辑运算，gpu执行矩阵运算 ②tf的分布式实现有哪些类别？单机单卡，单机多卡，多机多卡。
90630发布于 2018-09-25 - 来自专栏罗西的思考
并行分布式任务队列 Celery 之 Timer & Heartbeat
[源码分析] 并行分布式任务队列 Celery 之 Timer & Heartbeat 目录 [源码分析] 并行分布式任务队列 Celery 之 Timer & Heartbeat 0x00 摘要 Celery 之架构 (1) [源码解析] 并行分布式框架 Celery 之架构 (2) [源码解析] 并行分布式框架 Celery 之 worker 启动 (1) [源码解析] 并行分布式框架 Celery 之 worker 启动 (2) [源码解析] 分布式任务队列 Celery 之启动 Consumer [源码解析] 并行分布式任务队列 Celery 之 Task是什么 [从源码学设计]celery 之 发送Task & AMQP [源码解析] 并行分布式任务队列 Celery 之 消费动态流程 [源码解析] 并行分布式任务队列 Celery 之 多进程模型 [源码分析] 分布式任务队列 Celery 多线程模型 之 子进程 [源码分析]并行分布式任务队列 Celery 之 子进程处理消息 0x01 Blueprint Celery 的 Worker初始化过程中，其内部各个子模块的执行顺序是由一个
1.1K20发布于 2021-05-10
腾讯云开发者
Copyright © 2013 - 2026 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有
深圳市腾讯计算机系统有限公司 ICP备案/许可证号：粤B2-20090059 
粤公网安备44030502008569号
腾讯云计算（北京）有限责任公司 京ICP证150476号 | 京ICP备11018762号