logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

基于Atlas 800I A2的C4数据集下载与torchtitan 0.2.2训练指南

本文系统梳理了在昇腾NPU环境下使用torchtitan训练LLM时,基于C4数据集的完整部署流程。针对数据下载困难、环境配置复杂及训练启动效率低等问题,提出通过wget直接批量下载分片(推荐)替代hf命令行工具,并结合镜像站加速;详述了Atlas 800I A2 NPU的驱动、CANN、PyTorch及HCCL配置要点;强调预生成缓存的关键作用,有效规避多卡并行锁竞争导致的长时间无响应问题。附完

#python#pytorch
基于Atlas 800I A2的C4数据集下载与torchtitan 0.2.2训练指南

本文系统梳理了在昇腾NPU环境下使用torchtitan训练LLM时,基于C4数据集的完整部署流程。针对数据下载困难、环境配置复杂及训练启动效率低等问题,提出通过wget直接批量下载分片(推荐)替代hf命令行工具,并结合镜像站加速;详述了Atlas 800I A2 NPU的驱动、CANN、PyTorch及HCCL配置要点;强调预生成缓存的关键作用,有效规避多卡并行锁竞争导致的长时间无响应问题。附完

#python#pytorch
基于昇腾NPU的2D并行(FSDP+TP)计算流程解析:Transformer FFN层与Attention层

基于昇腾NPU的2D并行(FSDP+TP)计算流程解析:Transformer FFN层与Attention层

#python
基于昇腾NPU的2D并行(FSDP+TP)计算流程解析:Transformer FFN层与Attention层

基于昇腾NPU的2D并行(FSDP+TP)计算流程解析:Transformer FFN层与Attention层

#python
FSDP2 使用指南

随着大模型训练的普及,分布式训练技术成为提升训练效率、降低显存占用的关键手段。PyTorch 提供的 Fully Sharded Data Parallel(FSDP)是一种模型分片并行策略,能够将模型参数、梯度和优化器状态分布到多个设备上,从而支持训练超出单设备显存容量的大模型。FSDP2 是 FSDP 的下一代实现,基于 DTensor 进行参数管理,相比 FSDP1 在显存利用和计算效率上均

#python
FSDP2 使用指南

随着大模型训练的普及,分布式训练技术成为提升训练效率、降低显存占用的关键手段。PyTorch 提供的 Fully Sharded Data Parallel(FSDP)是一种模型分片并行策略,能够将模型参数、梯度和优化器状态分布到多个设备上,从而支持训练超出单设备显存容量的大模型。FSDP2 是 FSDP 的下一代实现,基于 DTensor 进行参数管理,相比 FSDP1 在显存利用和计算效率上均

#python
FSDP2 使用指南

随着大模型训练的普及,分布式训练技术成为提升训练效率、降低显存占用的关键手段。PyTorch 提供的 Fully Sharded Data Parallel(FSDP)是一种模型分片并行策略,能够将模型参数、梯度和优化器状态分布到多个设备上,从而支持训练超出单设备显存容量的大模型。FSDP2 是 FSDP 的下一代实现,基于 DTensor 进行参数管理,相比 FSDP1 在显存利用和计算效率上均

#python
到底了