自己做了一款在线 GPU 推理速度计算器 · TPS Calculator

4 月 27 日
 diudiuu

TPS Calculator · GPU 推理速度计算器

买不起机子,所以做了这个。

在线地址:tps.bunai.cc


突发奇想赶紧记录下来,直接 vibe code ,说敲就写

一个 vibe code 出来的 GPU 推理性能估算工具。

起因很简单——显卡太贵,买不起,想跑个模型又不知道自己的配置够不够, 于是把网上散落的参数和公式汇总了一下,做成了这个计算器。

输入显卡型号、模型、量化方式和运行参数,快速估算:


适合干什么

✅ 在买机子 / 租卡之前,先大概预估一下跑不跑得起来
✅ 学习推理性能建模,理解量化、KV Cache 、TP 、Roofline 这些概念
✅ 做方案初筛和参数对比

❌ 不适合直接替代真实 benchmark
❌ 不适合把估算值当作生产承诺
❌ Mac 电脑没有放出来,验证了一下差距有点大,先放一放


参考资料

这套公式和参数是我自己整理汇总的,没有大量真机跑过验证。 如果你手上有真实的测试数据,发现哪里估算偏差大、公式有问题, 欢迎开 Issue 或 PR 指出来,大家一起学习,一起把这个东西做得更准。

希望有真实数据的大佬帮忙指正,谢谢!🙏

1584 次点击
所在节点    Local LLM
12 条回复
qazwsxkevin
4 月 29 日
谢谢,可以增加 Qwen Coder 系列模型吗?
diudiuu
4 月 29 日
@qazwsxkevin 没问题的
diudiuu
4 月 29 日
@qazwsxkevin 已经追加了,希望能帮到你
diudiuu
4 月 29 日
@coefu 老哥看下小弟做的这个预测准不准
coefu
4 月 29 日
@diudiuu #4 其实对于这些细节,我也是一知半解的,我没沉下去细究过。😏

我去年囤了几条内存,今年出手赚了点差价,换了一个 amd mi50 32G ,最近都在折腾它。哎,一分钱一分货,早知道加点钱搞 v100 32G 了,折腾的太麻烦了。把我之前的主力机主板都搞崩了,就很烦。

你做的这个事情,很好,我目前帮不到什么忙。
nanshan2012
4 月 29 日
做得不错,提点建议。

在显存为 8G 的 4060 GPU 上跑 Qwen 3.6 35B MoE 模型,通过 offload 方式可以实现吞吐量在 20 token/s 以上,但网页提示的信息似乎有出入,请确认。
diudiuu
4 月 30 日
@nanshan2012
算法重新优化了一下,差不多

使用 MoE ,类型的需要默认打开 MoE CPU 卸载(在左下角),现在已经默认打开了

之前没有打开这个,没有考虑显存不够的情况,目前已经加上去了。

可以再试下
diudiuu
4 月 30 日
@coefu 等我拿这个赚钱,到时候买显卡
coefu
4 月 30 日
@diudiuu #8 太 tm 悲催了,我因为用错了电源线,把 amd mi50 的 hbm2 显存给烧了。以后坚决不买这种小众的东西了,电源线都找不到。还得是一分钱一分货。内存差价这种钱,果然还是赚不得。
nanshan2012
4 月 30 日
@diudiuu 不错不错,修正之后正常了,牛!!!
leven87
5 月 7 日
我的服务器是 NVIDIA GeForce RTX 4090 显卡, 但是本机检测不出来,要自己手动选。
diudiuu
5 月 7 日
@leven87 改了也不知道对不对,可以再试下

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/1208894

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX