PufferLib 5.0 单卡 880 万 SPS 训练魔方求解,性能惊人

haydendevs · x · 2026-11-11

开发者晒出 PufferLib 5.0 的训练界面截图:在单张 12GB 显卡上训练魔方(rubiks)环境,GPU 占满 100% 而显存仅用 1.8GB,达到 880 万 steps/秒(SPS)。

截图关键指标:2.8 亿步、epoch 1070、评估单步仅 621ms;perf 与 score 均达 1.000,最短求解 29 步,策略损失与 value 损失都趋近于 0。作者只配了「uhhhhh guys」一句,显然被自己库的性能吓到。PufferLib 是知名的极速强化学习训练库,这条截图本身就是极佳的性能展示。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →