AI 工作站算力规划
将以下提示词粘贴到你的 AI 对话框中:
请根据 https://skillhub.cn/install/skillhub.md,安装 @user_8d36cde0/guoyiyi-v052-azure-compute 技能到 AI 助手中。
技能介绍
要解决的具体问题
当工程师准备在本地部署 AI 工作站时,常遇到两类卡点:一是不清楚训练/推理目标对应的硬件底线,二是缺乏把预算拆解成 CPU、GPU、内存、存储等具体配置的规划方法。该技能围绕本地或机房内 AI 工作站场景,把"这台机器到底需要多少算力"变成可复核的规划过程,而不是凭经验随手堆配置。
核心能力与关键步骤
- 需求拆解:从任务类型、模型规模、批大小、延迟要求、数据量等参数出发,先明确工作站要承载的训练、微调或推理负载。
- 算力匹配:把负载映射到可用算力单元,评估 GPU/加速器、CPU 核心、内存带宽和容量、NVMe 吞吐是否构成合理瓶颈平衡,而不是单点过剩。
- 选型输出:给出可落地的配置方向,例如 GPU 数量与型号、内存配比、存储策略、电源与散热余量,并标注每项配置对应的假设。
- 风险提示:识别选型中的约束条件,如 PCIe/显存限制、跨卡通信需求、数据集是否适合本地缓存,以及扩展路径是否与当前架构兼容。
适用边界与注意点
- 该技能聚焦规划与决策支持,不替代真实基准测试;正式采购前仍需用代表性负载做 PoC 验证。
- 它不做云端实例比价,也不覆盖集群编排;如果目标是公有云训练集群,需要另外的选型方案。
- 输入参数越具体,输出越可靠;若只给模糊描述,结果会退化为通用参考。
- 最终硬件仍需检查主板接口、机箱风道、电源额定功率与散热设计,避免纸面配置无法落地。
使用场景
- 工程师准备搭建本地 AI 工作站,需要根据训练或推理负载确定 GPU、内存和存储配置。
- IT 团队在有限预算下规划 AI 工作站,希望平衡 CPU、GPU 和存储以避免单点瓶颈。
- 开发者需要为本地微调任务配置工作站,按模型规模和批大小评估显存与算力。
- 运维人员准备部署本地推理服务,依据延迟和并发要求给出工作站选型建议。
适合人员
- 准备自建 AI 工作站的工程师,需要把任务需求转换成可落地的硬件配置。
- 负责设备选型的 IT 运维人员,需要在预算约束下平衡算力、内存与存储。
- 本地部署大模型的开发者,需要评估训练、微调或推理所需的工作站规格。
- 支持 AI 研发的团队技术负责人,希望降低 AI 工作站规划与采购的沟通成本。