PyTorch
-
PyTorch 2.x分布式训练实战:DDP与FSDP并行方案对比
PyTorch 2.x分布式训练是深度学习框架落地大规模模型的基本功。单卡放不下模型时,DDP和FSDP是两条主流路径:DDP做数据并行,每张卡持有一份完整模型副本;FSDP把参数…
-
PyTorch FSDP分布式训练实战:全分片数据并行配置与显存优化
PyTorch FSDP分布式训练是什么 PyTorch FSDP(Fully Sharded Data Parallel)是PyTorch 2.0引入的全分片数据并行方案,将模型…