nvbench
nvbench 是一款专为简化 CUDA 内核性能测试而设计的 C++17 开源库。在 GPU 开发中,准确评估内核性能往往面临配置空间复杂、环境变量干扰大等挑战,nvbench 正是为了解决这些痛点而生。它帮助开发者轻松构建基准测试,自动探索不同的参数组合,并提供精确的吞吐量(如每秒元素数、显存带宽利用率)及执行时间统计。
这款工具特别适合 CUDA 开发者、高性能计算研究人员以及需要优化 GPU 算法的工程师使用。其核心亮点在于强大的“参数轴”系统,支持动态数值、字符串乃至静态类型的灵活扫描,让用户能全面覆盖内核的配置空间。此外,nvbench 提供了丰富的命令行接口,允许用户在运行时自定义测试参数、锁定 GPU 频率或切换输出格式(Markdown/CSV),无需重新编译代码。它还支持多种测量模式,包括清除缓存后的“冷启动”测试、批量平均测试以及纯 CPU 耗时测量,并可选手动控制计时起止,确保数据真实可靠。通过几行简洁的代码,用户即可生成专业的性能报告,显著提升 GPU 内核的调试与优化效率。
使用场景
某高性能计算团队正在为新一代图像处理算法优化 CUDA 内核,需要在不同数据规模和 GPU 架构下验证性能表现。
没有 nvbench 时
- 开发人员需手动编写繁琐的测试循环来遍历不同的线程块大小和数据量,代码复用性极差且容易出错。
- 缺乏统一的指标统计,每次运行后需人工计算吞吐量(如元素/秒)和显存带宽利用率,难以快速判断瓶颈。
- 无法灵活地在运行时切换测试参数或锁定 GPU 频率,导致在不同环境下的对比实验数据一致性难以保证。
- 测试结果分散在日志文件中,整理成可视化的 Markdown 表格或 CSV 报告需要耗费大量额外时间进行后处理。
使用 nvbench 后
- 利用强大的“参数轴”系统,仅需几行配置即可自动扫描多种内核配置空间,轻松实现动态参数与静态类型的组合测试。
- 内置吞吐量计算功能,自动输出每项测试的元素处理速度和显存带宽占用百分比,直观展示性能水位。
- 通过丰富的命令行接口,无需重新编译即可动态重定义测试参数、指定设备或锁定时钟频率,确保实验条件严格可控。
- 直接生成格式规范的 Markdown 或 CSV 报告,清晰呈现冷启动与批量执行的对比数据,极大加速了性能分析报告的产出。
nvbench 将原本耗时数天的内核性能调优与数据整理工作缩短至小时级,让开发者能更专注于算法逻辑本身的优化。
运行环境要求
- 未说明
必需 NVIDIA GPU (支持 Volta 架构及以上以锁定时钟),需安装 CUDA Toolkit 12.0+,具体显存大小取决于基准测试配置
未说明

快速开始
概述
本项目目前仍在开发中,所有内容均可能发生变化。
NVBench 是一个基于 C++17 的库,旨在简化 CUDA 内核的基准测试。其主要特性包括:
- 参数扫描:强大的灵活“轴”系统可以探索内核的配置空间。参数可以是动态的数字或字符串,也可以是静态类型。
- 运行时自定义:丰富的命令行界面允许重新定义参数轴、选择 CUDA 设备、锁定 GPU 时钟(Volta 及以上)、更改输出格式等。
- 吞吐量计算:计算并报告:
- 项吞吐量(元素/秒)
- 全局内存带宽使用情况(字节/秒及占峰值带宽的百分比)
- 多种输出格式:目前支持 Markdown(默认)和 CSV 格式。
- 手动计时模式:(可选)在基准测试实现中显式地开始/停止计时。
- 多种测量类型:
- 冷启动测量:
- 每个样本都会在设备 L2 缓存清空的情况下运行一次基准测试。
- 报告 GPU 和 CPU 时间。
- 批量测量:
- 连续多次执行基准测试并记录总时间。
- 报告平均执行时间(总时间 / 执行次数)。
- 仅 CPU 测量
- 测量非 GPU 基准测试的主机端执行时间。
- 不适用于微基准测试。
- 冷启动测量:
请观看这场演讲,了解 CUDA 内核基准测试固有的挑战,以及 NVBench 如何为您解决这些问题!
支持的编译器和工具
- CMake > 3.30.4
- CUDA 工具包 + nvcc:12.0 及以上版本
- g++:7 至 14 版本
- clang++:14 至 19 版本
- 头文件已测试支持 C++17 至 C++20。
快速入门
最简基准测试
只需几行 CUDA C++ 代码即可创建一个基本的内核基准测试:
void my_benchmark(nvbench::state& state) {
state.exec([](nvbench::launch& launch) {
my_kernel<<<num_blocks, 256, 0, launch.get_stream()>>>();
});
}
NVBENCH_BENCH(my_benchmark);
有关自定义基准测试和实现参数扫描的信息,请参阅基准测试。
命令行界面
NVBench 生成的每个基准测试可执行文件都提供丰富的命令行选项,用于在运行时配置基准测试的执行。更多信息请参阅CLI 概览和CLI 轴指定。
示例
此仓库提供了多个示例,展示了 NVBench 的各种功能和用例:
- 运行时和编译时参数扫描
- 仅 CPU 基准测试
- 枚举和编译时常量整数参数轴
- 报告项/秒和字节/秒的吞吐量统计信息
- 跳过某些基准配置
- 在特定流上进行基准测试
- 在 Markdown 输出中添加或隐藏列(摘要)
- 同步 CUDA 设备的基准测试:
nvbench::exec_tag::sync - 手动计时:
nvbench::exec_tag::timer
构建示例
要构建示例:
mkdir -p build
cd build
cmake -DNVBench_ENABLE_EXAMPLES=ON -DCMAKE_CUDA_ARCHITECTURES=70 .. && make
请确保根据您正在使用的 GPU 设置 CMAKE_CUDA_ARCHITECTURE。
示例默认会构建到 build/bin 目录下,并以 nvbench.example 为前缀。
来自 `nvbench.example.throughput` 的示例输出
# 设备
## [0] `Quadro GV100`
* SM 版本:700(PTX 版本:700)
* SM 数量:80
* SM 默认主频:1627 MHz
* 全局内存:32163 MiB 空闲 / 32508 MiB 总容量
* 全局内存总线峰值:870 GiB/秒(4096 位 DDR @850MHz)
* 最大共享内存:96 KiB/SM,48 KiB/块
* L2 缓存大小:6144 KiB
* 最大活动块数:32/SM
* 最大活动线程数:2048/SM,1024/块
* 可用寄存器:65536/SM,65536/块
* ECC 是否启用:否
# 日志
运行: throughput_bench [设备=0]
警告:当前测量超时(15.00s),且噪声超过阈值(1.26% > 0.50%)
通过:冷启动:0.262392ms GPU,0.267860ms CPU,7.19s 总 GPU 时间,27393 次
通过:批量:0.261963ms GPU,总 GPU 时间 7.18s,27394 次
# 基准测试结果
## throughput_bench
### [0] Quadro GV100
| NumElements | DataSize | Samples | CPU Time | Noise | GPU Time | Noise | Elem/s | GlobalMem BW | BWPeak | Batch GPU | Batch |
|-------------|------------|---------|------------|-------|------------|-------|---------|---------------|--------|------------|--------|
| 16777216 | 64.000 MiB | 27393x | 267.860 us | 1.25% | 262.392 us | 1.26% | 63.940G | 476.387 GiB/s | 58.77% | 261.963 us | 27394x |
演示项目
要开始使用 NVBench 对您自己的内核进行基准测试,可以尝试NVBench 演示项目。
nvbench_demo 提供了一个简单的 CMake 项目,使用 NVBench 构建示例基准测试。这是一个无需过多投入即可体验该库的绝佳方式。
贡献
欢迎贡献!
有关当前问题,请查看问题板。标有 的问题非常适合首次贡献者。
测试
要构建 nvbench 测试:
mkdir -p build
cd build
cmake -DNVBench_ENABLE_TESTING=ON .. && make
测试默认会构建到 build/bin 目录下,并以 nvbench.test 为前缀。
运行所有测试:
make test
或者
ctest
许可证
NVBench 采用 Apache 2.0 许可证,并附带 LLVM 异议条款。 详情请参阅LICENSE。
范围及相关项目
NVBench 将为每个基准测试测量单个主机端关键区域的 CPU 和 CUDA GPU 执行时间。它主要用于单个内核的回归测试和参数调优。对于多应用端到端性能的深入分析,NVIDIA Nsight 工具更为合适。
NVBench 专注于评估 CUDA 内核的性能。它还提供仅 CPU 基准测试功能,适用于非平凡的 CPU 工作负载,但并不针对 CPU 微基准测试进行优化。未来可能会有所改变,但目前建议使用 Google Benchmark 来进行高分辨率的 CPU 基准测试。
版本历史
python-0.2.02026/02/04python-0.1.02026/01/30常见问题
相似工具推荐
ML-For-Beginners
ML-For-Beginners 是由微软推出的一套系统化机器学习入门课程,旨在帮助零基础用户轻松掌握经典机器学习知识。这套课程将学习路径规划为 12 周,包含 26 节精炼课程和 52 道配套测验,内容涵盖从基础概念到实际应用的完整流程,有效解决了初学者面对庞大知识体系时无从下手、缺乏结构化指导的痛点。 无论是希望转型的开发者、需要补充算法背景的研究人员,还是对人工智能充满好奇的普通爱好者,都能从中受益。课程不仅提供了清晰的理论讲解,还强调动手实践,让用户在循序渐进中建立扎实的技能基础。其独特的亮点在于强大的多语言支持,通过自动化机制提供了包括简体中文在内的 50 多种语言版本,极大地降低了全球不同背景用户的学习门槛。此外,项目采用开源协作模式,社区活跃且内容持续更新,确保学习者能获取前沿且准确的技术资讯。如果你正寻找一条清晰、友好且专业的机器学习入门之路,ML-For-Beginners 将是理想的起点。
funNLP
funNLP 是一个专为中文自然语言处理(NLP)打造的超级资源库,被誉为"NLP 民工的乐园”。它并非单一的软件工具,而是一个汇集了海量开源项目、数据集、预训练模型和实用代码的综合性平台。 面对中文 NLP 领域资源分散、入门门槛高以及特定场景数据匮乏的痛点,funNLP 提供了“一站式”解决方案。这里不仅涵盖了分词、命名实体识别、情感分析、文本摘要等基础任务的标准工具,还独特地收录了丰富的垂直领域资源,如法律、医疗、金融行业的专用词库与数据集,甚至包含古诗词生成、歌词创作等趣味应用。其核心亮点在于极高的全面性与实用性,从基础的字典词典到前沿的 BERT、GPT-2 模型代码,再到高质量的标注数据和竞赛方案,应有尽有。 无论是刚刚踏入 NLP 领域的学生、需要快速验证想法的算法工程师,还是从事人工智能研究的学者,都能在这里找到急需的“武器弹药”。对于开发者而言,它能大幅减少寻找数据和复现模型的时间;对于研究者,它提供了丰富的基准测试资源和前沿技术参考。funNLP 以开放共享的精神,极大地降低了中文自然语言处理的开发与研究成本,是中文 AI 社区不可或缺的宝藏仓库。
cs-video-courses
cs-video-courses 是一个精心整理的计算机科学视频课程清单,旨在为自学者提供系统化的学习路径。它汇集了全球知名高校(如加州大学伯克利分校、新南威尔士大学等)的完整课程录像,涵盖从编程基础、数据结构与算法,到操作系统、分布式系统、数据库等核心领域,并深入延伸至人工智能、机器学习、量子计算及区块链等前沿方向。 面对网络上零散且质量参差不齐的教学资源,cs-video-courses 解决了学习者难以找到成体系、高难度大学级别课程的痛点。该项目严格筛选内容,仅收录真正的大学层级课程,排除了碎片化的简短教程或商业广告,确保用户能接触到严谨的学术内容。 这份清单特别适合希望夯实计算机基础的开发者、需要补充特定领域知识的研究人员,以及渴望像在校生一样系统学习计算机科学的自学者。其独特的技术亮点在于分类极其详尽,不仅包含传统的软件工程与网络安全,还细分了生成式 AI、大语言模型、计算生物学等新兴学科,并直接链接至官方视频播放列表,让用户能一站式获取高质量的教育资源,免费享受世界顶尖大学的课堂体验。
ragflow
RAGFlow 是一款领先的开源检索增强生成(RAG)引擎,旨在为大语言模型构建更精准、可靠的上下文层。它巧妙地将前沿的 RAG 技术与智能体(Agent)能力相结合,不仅支持从各类文档中高效提取知识,还能让模型基于这些知识进行逻辑推理和任务执行。 在大模型应用中,幻觉问题和知识滞后是常见痛点。RAGFlow 通过深度解析复杂文档结构(如表格、图表及混合排版),显著提升了信息检索的准确度,从而有效减少模型“胡编乱造”的现象,确保回答既有据可依又具备时效性。其内置的智能体机制更进一步,使系统不仅能回答问题,还能自主规划步骤解决复杂问题。 这款工具特别适合开发者、企业技术团队以及 AI 研究人员使用。无论是希望快速搭建私有知识库问答系统,还是致力于探索大模型在垂直领域落地的创新者,都能从中受益。RAGFlow 提供了可视化的工作流编排界面和灵活的 API 接口,既降低了非算法背景用户的上手门槛,也满足了专业开发者对系统深度定制的需求。作为基于 Apache 2.0 协议开源的项目,它正成为连接通用大模型与行业专有知识之间的重要桥梁。
PaddleOCR
PaddleOCR 是一款基于百度飞桨框架开发的高性能开源光学字符识别工具包。它的核心能力是将图片、PDF 等文档中的文字提取出来,转换成计算机可读取的结构化数据,让机器真正“看懂”图文内容。 面对海量纸质或电子文档,PaddleOCR 解决了人工录入效率低、数字化成本高的问题。尤其在人工智能领域,它扮演着连接图像与大型语言模型(LLM)的桥梁角色,能将视觉信息直接转化为文本输入,助力智能问答、文档分析等应用场景落地。 PaddleOCR 适合开发者、算法研究人员以及有文档自动化需求的普通用户。其技术优势十分明显:不仅支持全球 100 多种语言的识别,还能在 Windows、Linux、macOS 等多个系统上运行,并灵活适配 CPU、GPU、NPU 等各类硬件。作为一个轻量级且社区活跃的开源项目,PaddleOCR 既能满足快速集成的需求,也能支撑前沿的视觉语言研究,是处理文字识别任务的理想选择。
awesome-machine-learning
awesome-machine-learning 是一份精心整理的机器学习资源清单,汇集了全球优秀的机器学习框架、库和软件工具。面对机器学习领域技术迭代快、资源分散且难以甄选的痛点,这份清单按编程语言(如 Python、C++、Go 等)和应用场景(如计算机视觉、自然语言处理、深度学习等)进行了系统化分类,帮助使用者快速定位高质量项目。 它特别适合开发者、数据科学家及研究人员使用。无论是初学者寻找入门库,还是资深工程师对比不同语言的技术选型,都能从中获得极具价值的参考。此外,清单还延伸提供了免费书籍、在线课程、行业会议、技术博客及线下聚会等丰富资源,构建了从学习到实践的全链路支持体系。 其独特亮点在于严格的维护标准:明确标记已停止维护或长期未更新的项目,确保推荐内容的时效性与可靠性。作为机器学习领域的“导航图”,awesome-machine-learning 以开源协作的方式持续更新,旨在降低技术探索门槛,让每一位从业者都能高效地站在巨人的肩膀上创新。