独立完成,复现GPT-2 (124M参数)+FP8混合精度预训练
● 基于Transformer架构
● 训练数据是HuggingFaceFW/fineweb-edu中的sample-10BT
● GPU租用的是智星云,GeForce RTX 4090 (24G),系统Ubuntu,单机4卡
● 相关版本如下:CUDA Version: 12.4, Driver Version: 550.127.05, torch 2.5.0, triton 3.1.0
需求方专属客服,免费梳理匹配
添加客服微信,免费为您安排与该工程师直接沟通
长按二维码添加客服微信
如当前档期不合,也可免费为您推荐相似案例作者