PyTorch 实测:批量从 1 加到 64,吞吐升 50 倍延迟仅增 24%

cneuralnetwork · x · 2026-10-23

作者在 PyTorch 2.8.0(MPS 设备)上对小神经网络做基准测试,直观展示了 batching 的吞吐-延迟权衡:batch size 从 1 提到 32 时单批延迟稳定在 0.076ms,吞吐从 13,178 req/s 线性升到 419,863 req/s;batch 64 时延迟才升至 0.094ms,吞吐达 680,811 req/s。原因在于 GPU 为并行计算设计:小批量会让大量算力闲置,大批量提高利用率,但换来的是整体吞吐而非单个请求的延迟改善。

所属事件:PyTorch 实测 batching:吞吐大涨而延迟仅微增(2 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →