PyTorch 2.15 LU 求解器性能暴跌百倍,作者劝勿升级

ducha_aiki · x · 2026-11-12

研究者 duchaaiki 警告不要升级 PyTorch 2.15:LU 分解性能严重回退,最坏情况慢约 100 倍。其附带的复现数据显示:在 RTX 4090 上对 8192 个 (9,8) float32 矩阵做 torch.linalg.lufactorex,2.14.0+cu130 默认后端仅 8ms,而 2.15.0+cu132 达 924ms;2.14 会自动选择批量 MAGMA kernel,2.15 起所有后端(含强制 MAGMA)都退化为循环速度。下游影响实测:Kornia 的 RANSAC 中基础矩阵调用从 42ms 涨到 1.08s,单应性调用从 125ms 涨到 0.98s,某些设置下甚至比 CPU 还慢 3.5-80 倍。作者确认问题源于 2.14 到 2.15 之间的代码变更而非 CUDA 13 构建。

原文链接 →

「编程与Agent」频道最新

更多「编程与Agent」频道 AI 资讯 →