superbenchmark

一键体检 AI 集群，跑分定位硬件瓶颈

SuperBenchmark 是微软开源的 AI 基础设施验证与性能分析工具，面向 GPU 集群、云上 AI 算力平台和自建训练环境。它解决的核心问题是：在部署大模型训练或推理之前，如何快速确认硬件、驱动、网络、存储等基础设施是否正常，并量化其性能表现。工具通过模块化的基准测试套件，覆盖 GPU 计算、显存带宽、NVLink/PCIe 通信、RDMA 网络、CPU、内存、磁盘 IO 等维度，自动生成可对比的报告。它支持 Azure、本地裸金属和 Kubernetes 环境，可集成到 CI/CD 或集群验收流程中，帮助运维和算法团队在故障发生前定位瓶颈，避免因硬件配置不当导致训练中断或性能损失。项目采用 Python 编写，提供命令行接口和配置化测试集，便于扩展自定义基准。