讲师中心 微信公众号
AI工具推荐 视频效率加速

本地大模型推理速度测试_每秒Token生成数量

酷杰大大_4696

酷杰大大_4696

发布时间:2026-04-28 13:54:47

|

1010人浏览过

|

来源于php中文网

原创

评估本地大模型吞吐能力的关键指标是每秒生成Token数(TPS),可通过三种方法测量:一、用llm-perf工具标准化测试,固定输入输出长度并统计平均TPS;二、通过vLLM API响应头中的生成耗时与token数反推瞬时TPS;三、用transformers加torch.compile手动计时,同步GPU后计算端到端TPS。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 多模态理解力帮你轻松跨越从0到1的创作门槛☜☜☜

本地大模型推理速度测试_每秒token生成数量

如果您希望评估本地部署的大模型在实际推理过程中的吞吐能力,关键指标之一是每秒生成的Token数量(Tokens per Second, TPS)。该数值直接受限于硬件配置、模型量化方式、推理框架优化程度及输入输出长度。以下是测量本地大模型推理速度的具体步骤:

一、使用llm-perf工具进行标准化基准测试

llm-perf是一个专为大语言模型设计的轻量级性能评测工具,支持统一上下文长度、批量大小与采样参数,可排除prompt解析和后处理环节干扰,专注测量核心生成阶段的TPS。

1、克隆llm-perf仓库并进入项目目录:git clone https://github.com/InternLM/llm-perf && cd llm-perf。

2、安装依赖并指定目标模型路径:pip install -r requirements.txt && python setup.py install。

3、执行单轮生成测试,固定输入长度为512,输出长度为256:python benchmark.py --model-path /path/to/your/model --input-len 512 --output-len 256 --batch-size 1。

4、工具将自动运行10次warmup后采集50次有效推理,最终输出平均TPS值及标准差。

二、通过vLLM服务端API日志提取实时TPS

vLLM提供高吞吐的PagedAttention推理后端,其HTTP API在响应头中嵌入了详细的时序信息,包括token生成耗时,可用于反向计算真实TPS。

1、启动vLLM服务并启用详细日志:python -m vllm.entrypoints.api_server --model /path/to/your/model --enable-prefix-caching --log-level DEBUG。

2、发送一次包含128个输入token、请求生成200个输出token的请求,并记录完整响应体与响应头。

大模型Token成本节约
大模型Token成本节约

大模型 Token 成本节约工具。在请求到达大模型之前自动压缩 prompt 和上下文,减少 60-95% 的 token 消耗,直接降低 API 成本。支持 Claude/OpenAI/Gemini 等主流模型,提供代理模式、CLI 包装、Python SDK 和 MCP Server 四种接入方式。内置一键安...

下载

3、从响应头中提取x-vllm-generation-time-ms与x-vllm-num-generated-tokens字段值。

4、用生成token总数除以生成耗时(单位转换为秒),得出本次请求的瞬时TPS:TPS = num_generated_tokens / (generation_time_ms / 1000)。

三、利用transformers + torch.compile手动计时

在不依赖专用推理框架的前提下,可通过原生Hugging Face Transformers加载模型,结合PyTorch 2.0编译机制与精确时间戳,获得端到端生成阶段的TPS数据。

1、加载模型时启用torch.compile并设置动态shape支持:model = torch.compile(model, dynamic=True, mode="reduce-overhead")。

2、禁用所有非必要输出,仅保留logits并强制同步GPU:torch.cuda.synchronize(); start_time = time.time()。

3、执行generate调用,设置max_new_tokens=256且do_sample=False以消除采样开销。

4、再次同步GPU并记录结束时间:torch.cuda.synchronize(); end_time = time.time()。

5、计算TPS:用256除以(end_time - start_time),结果即为本次生成的实测TPS。

热门AI工具

更多
Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

相关专题

更多
Uvicorn 报错与问题解决教程
Uvicorn 报错与问题解决教程

本专题汇总了Uvicorn服务器常见问题与报错解决的教程合集,涵盖启动失败、端口占用、模块导入错误、事件循环冲突及热重载失效等典型故障的排查思路。同时整理了多进程与Worker配置、SSL/HTTPS证书报错、跨域与代理配置等实战解决方案,助你快速定位并解决Uvicorn运行中的各类问题。

0

2026.11.02

Uvicorn使用教程
Uvicorn使用教程

本专题汇总了Uvicorn使用教程与启动运行指南,涵盖其作为Python ASGI服务器的定位与特点,以及命令行启动(uvicorn main:app --reload)、参数配置(host/port/workers)、程序内嵌运行与Gunicorn结合部署等核心用法。同时整理了热重载调试、日志配置与生产环境部署优化等技巧,助你快速掌握Uvicorn的运行与配置。

0

2026.11.02

Uvicorn安装配置教程
Uvicorn安装配置教程

本专题汇总了Uvicorn的安装配置教程与部署指南,涵盖通过pip安装Uvicorn、启动FastAPI/Starlette应用、开发模式热重载(--reload)及多进程工作模式配置等核心操作。同时整理了结合Gunicorn的生产部署方案、HTTPS与反向代理配置及性能调优技巧,助你快速完成Uvicorn服务的安装与上线部署。

0

2026.11.02

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

120

2026.10.28

Vite 如何搭配 Vue.js
Vite 如何搭配 Vue.js

Vite搭配Vue.js专题汇集创建Vue 3项目、配置开发环境、路径别名、Vue插件、代理、多环境构建与调试的已审核文章,帮助读者完成从初始化到日常开发的配置。

80

2026.10.28

Vite 项目目录怎么看
Vite 项目目录怎么看

Vite项目目录专题汇集入口文件、src源码、public静态资源、路径别名、项目根目录和构建输出目录相关的已审核文章,帮助读者理解各文件夹的用途与引用方式。

280

2026.10.28

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

160

2026.10.22

Gradle Wrapper配置教程
Gradle Wrapper配置教程

本专题介绍Gradle Wrapper的基本作用,涵盖文件组成、生成方法、版本配置和任务调用,帮助你理解项目如何使用指定的Gradle版本,减少不同开发环境带来的构建差异。

140

2026.10.22

Golangci-lint检查器使用教程
Golangci-lint检查器使用教程

本专题整理Golangci-lint常用Linters的用途与配置方法,覆盖错误处理、无效代码、安全、复杂度、命名和依赖检查,帮助项目合理选规则。

100

2026.10.22

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn