Distributed data parallel (DDP)分布式训练

一次搞懂PyTorch DDP分布式训练_哔哩哔哩_bilibili

多机多卡,或单机多卡

几张卡初始化模型一样,每张卡分别拿不同的数据

几张开训练一轮后,会共享自己反向传播的梯度,然后对梯度取均值,来达到同步跟新的效果

all reduce:就是把所有卡的信息收即到自己这里来的这么一个通信操作。(4张卡就4个都会做,且是对同一批参数梯度同时对齐,也只有这样才能同时对齐)

(注意他同步梯度的时候,是边反向传播,边互相同步,而不是等反向传播完了之后在同步)

靠这个随机种子,来保证取数据大家 是同步的

jakeandjane/ddp-demo: Pytorch DDP Traning Demo

相关推荐
敖正炀13 小时前
高并发系统的降级预案与容错策略
分布式·架构
敖正炀14 小时前
稳定性监控与告警体系:SLI/SLO/SLA 实践
分布式·架构
敖正炀14 小时前
故障演练与混沌工程:ChaosBlade 到 Litmus
分布式·架构
敖正炀14 小时前
全链路压测与容量规划方法论
分布式·架构
敖正炀14 小时前
限流算法深度与 Guava/Sentinel 源码:从单机令牌桶到分布式滑动窗口的流量防护体系
分布式·架构
山屿落星辰18 小时前
hixl - 让分布式训练“零拷贝“通信
分布式
逍遥德20 小时前
SpringBoot自带TaskScheduler 接口使用详解:(02)微服务多实例模式下,爆发任务重复执行问题
spring boot·分布式·后端·微服务·中间件
Solis程序员21 小时前
基于 Outbox 事务表 + Canal 监听+kafka+多级缓存:高并发社交关注系统全链路架构设计
分布式·kafka·linq
phltxy21 小时前
Redis集群:分布式高可用存储方案
数据库·redis·分布式
二宝哥21 小时前
大数据之安装zookeeper
大数据·分布式·zookeeper