国产AI芯片选型指南:训练算力与推理专用芯片全解析 做AI基础设施这几年被问得最多的不是“怎么把模型效果调上去”而是“预算有限算力芯片到底买哪家”。加上国产AI综合生态这两年的热度算力芯片和推理专用芯片这两个词几乎成了每个技术人绕不开的考题。这篇文章我想用做实事的口吻把国产AI芯片厂商按训练算力和推理专用两条线拆开讲再做一份可以直接参考的分类推荐清单重点说明不同场景怎么选、迁移时会踩哪些坑。无论你是在搭训练集群还是准备把大模型上线做推理都建议先看完再动手。1. 算力芯片与推理专用芯片到底是不是两种东西1.1 训练和推理的底层逻辑不同先说结论训练芯片和推理专用芯片在硬件设计上就是两个物种不是“一个贵一个便宜”的关系。训练的本质是反复迭代。模型参数从随机初始化开始通过正向传播算损失、反向传播更新权重一个batch一个batch地磨。这个过程对芯片的诉求是并行计算能力拉满大矩阵乘法和卷积要快显存尽量大数据带宽尽量高同时要支持足够的精度FP16、BF16是主流。你可以把训练想象成盖一栋楼工人越多、建材越多、塔吊越快工期就压得越短。推理的本质是服务。模型已经训练好了参数固定要做的只是对新输入数据做一次前向计算输出结果。这时候计算量的绝对值往往没训练大但业务要求的是低延迟、高吞吐、低功耗。推理专用芯片更像大楼交付之后的物业系统单个活儿不大但响应要快、要稳定、单位电费里要能服务尽可能多的人。这导致了两者的设计取向完全不同通用训练芯片追求极致浮点算力和灵活的指令集什么模型都能算推理专用芯片则可以把某些固定形态的计算比如Transformer里的Attention、FeedForward做成硬加速模块牺牲灵活性换效率。所以不能用一块训练卡硬扛所有推理也不能指望推理芯片帮你训大模型。1.2 算力和推理的核心指标差异判断一块芯片到底是训练向还是推理向有四个指标最直观算力单位训练通常讲FP16、BF16下的TFLOPS推理常用INT8、INT4下的TOPS因为推理模型中量化部署是常态。显存/HBM带宽训练吃显存容量和带宽权重、梯度、优化器状态都要驻留推理虽然也吃显存但更看重带宽对token生成速度的影响。互联能力训练多卡并行必须依赖高速互联PCIe都不够用得看HCCS、NVLink这类专用互联推理往往单卡或多卡负载均衡就够。软件栈训练芯片的软件栈要覆盖分布式训练框架推理芯片则要重点看推理引擎、量化工具、服务框架的成熟度。如果一张卡标称FP16算力很高、互联带宽很强那大概率是训练卡如果标称INT8算力很高且专门宣传了Attention加速、PagedAttention支持那基本是推理专用卡。这个判断逻辑在国产芯片里同样适用。1.3 为什么现在国产芯片值得认真看早几年做AI选型基本是闭眼选进口卡不是因为它多好而是因为生态省心。但现在情况变了我把原因归纳成三点。第一算力供给的确定性。芯片采购周期越来越长排队几个月是常事纯进口方案的供应链风险太高谁也不想辛辛苦苦搭完集群结果产能跟不上。第二推理业务爆发。大模型从尝鲜走向生产推理成本成了企业最头疼的账单而推理恰好是国产芯片最能发挥性价比优势的战场。第三生态成熟度上来了。头部国产厂商的算子库、编译框架、分布式支持已经能覆盖大部分业务模型虽然迁移还是有成本但不再是“完全不能用”。说白了研究国产AI芯片已经不是情怀题而是实打实的成本工程和稳定性工程早一天把这件事想清楚早一天掌握主动权。2. 国产AI算力芯片厂商全景分类推荐2.1 云端训练算力芯片梯队先看训练这条线目前拿得出训练产品的厂商并不少但梯队差距很明显。华为昇腾是绕不开的第一名。昇腾910B及后续型号在国产训练卡里算是综合实力最强的一档FP16算力处于百TFLOPS级别HCCS互联在四卡、八卡场景下表现稳定支持FP16、BF16以及混合精度训练。如果想跑PyTorch可以直接用CANN适配社区案例比较多。代价是软件栈与CUDA生态不兼容迁移代码不是无痛的。寒武纪的思元系列也是老牌选手。思元590这一代强化了训练能力指令集设计灵活适合喜欢深度定制算子、又不想被厂商框架绑架的团队。但它的开源生态和中文资料相对少外部开发者上手周期比昇腾长。海光信息走的是x86兼容路线DCU在迁移性上有天然优势。它的优势很实在如果你手里是一堆成熟的CUDA代码海光有工具链可以做自动转换转换完再人工修工作量大为降低。对于“存量代码为主、又要求国产化落地”的企业海光这条路值得优先调研。新势力里沐曦的曦云系列值得关注。它的架构是类CUDA设计生态兼容思路跟海光类似同时在FP16算力和显存配置上堆料很足。但目前大规模集群的稳定性验证案例还不够多适合做技术验证。壁仞和摩尔线程也都在做训练卡前者纸面规格高问题在驱动成熟度后者消费端起家计算卡典型跑中小任务更靠谱。2.2 推理专用芯片赛道选手推理专用是国产芯片竞争最激烈、也最贴近业务价值的赛道。百度昆仑芯落地案例最丰富。昆仑芯R200系列针对大语言模型推理做了专门设计对Attention计算加速明显配合PaddlePaddle和自研推理引擎生成式任务的表现很能打。而且昆仑芯对ONNX模型支持不错业务方把训练好的PyTorch模型导出为ONNX就能跑起来迁移成本低。阿里含光800是ASIC路线编译器针对搜索、推荐类模型做了深度优化在阿里内部业务中验证充分。但它主要是为阿里云自用设计外部团队想直接采购和部署路径不算开放。如果你在阿里云上做推理可以考虑云上实例能力。燧原科技的云燧i20系列专注云端推理INT8算力与单位功耗的性价比做得不错软件栈对Transformer类模型做了不少优化。天数智芯的智铠系列主打边缘和推理场景在视频分析、图像识别类任务上性价比突出。这两家更适合对成本敏感的创业团队做点对点适配。2.3 边缘与端侧国产AI芯片端侧和边缘是另一个赛道核心诉求是低功耗、低成本和不依赖数据中心。地平线的征程系列主要面向车载和机器人黑芝麻的华山系列也聚焦车规级场景它们把NPU和图像处理结合得很紧。如果做的是智能摄像头、边缘盒子、智能座舱这类产品瑞芯微等厂商的NPU方案量很大、成本很好控。端侧选芯片我的经验是不用盯着TOPS数字更多的要看在目标功耗下跑你的具体模型能不能达到要求的帧率和精度——这也是端侧和云端选型最大的差异。3. 选型决策不同业务场景该怎么挑芯片3.1 训练场景的选型逻辑训练场景选型我的经验是第一看生态第二看互联第三才看纸面算力。生态决定了你的迁移成本。团队主力是PyTorch就重点考察目标芯片对PyTorch的支持程度如果愿意换旗舰生态昇腾CANN的组合在稳定性上更优。注意这里的“支持”不只是能跑而是要看正统的算子和内建函数覆盖率。你至少要把模型里用得最多的十几个算子列出来逐个确认对方算子库支持不支持、有没有性能基准。然后是互联。多卡训练如果互联带宽不行加再多卡也白搭。实际调研时不要只看单卡指标一定要看四卡或八卡的扩展效率测试数据最好是自己测拿同一个模型在1卡和4卡上跑看加速比能达到多少。很多国产卡的纸面四卡加速比是3.8倍实测可能只有2.5倍甚至更低数据一测就现形。最后才是算力参数。FP16算力、显存容量、显存带宽三者要一起看比如跑一个70B量级模型微调显存直接决定能不能塞得下带宽决定训练速度天花板。3.2 推理场景的选型逻辑推理选型和训练是完全不同的思路。核心指标不是TFLOPS而是四个词延迟、吞吐、成本、稳定性。延迟要看首token延迟和decode阶段每token的生成时间这两个数据最能反映芯片对Transformer的优化程度。吞吐看单卡能支撑多少并发请求跟显存容量和batch策略强相关。成本要用TCO来看不是采购价而是把功耗、机房、运维、软件适配都算进去后分摊到每一次推理调用的成本。推理专用芯片的一大优势在量化能力。FP16转INT8、INT4后如果工具链成熟模型体积直接缩小推理速度明显提升。挑选时务必问清楚你们支持哪些量化方法GPTQ、AWQ是否支持校准工具是否好用这决定了你的模型能不能在国产推理卡上跑出理想的性价比。3.3 一个可以抄作业的小型验证流程不管芯片厂商宣传得多么好看我建议动手前先做一个“最小可行性验证”流程如下第一步准备一个你最典型的业务模型最好带上你线上真实的数据样本第二步在目标芯片上完成模型转换记录转换工具的报错情况、算子缺失数量、手动修复工作量第三步跑一轮小规模推理基准记录延迟和吞吐如果是训练卡就跑1卡和4卡各一轮训练记录加速比第四步用同样的数据在你在用的现有芯片上跑一遍得到对比基线第五步把结果做成一张表按“适配工作量、性能差距、成本节省、稳定性风险”四个维度打分。这个过程大概两到三周能完成但它比看任何厂商白皮书都靠谱。选型最大的风险就是“按参数表买卡”参数表是理想情况业务模型才是现实情况。4. 实操经验国产芯片迁移中的账本与坑4.1 生态迁移的真实时间账分享一下我做过的一次真实迁移案例。一个中小规模的对话模型架构偏Transformer原本跑在CUDA平台上负责的团队有两个算法工程师和一名运维。我们选择迁移到一家国产NPU训练平台整个过程用了大约一个半月才跑通再花了三周做性能调优前后将近两个月超过了我最初的预估。时间都花在哪了算子适配是大头大约占50%因为模型里有一些自定义Op对方算子库不直接支持需要写shader或者调用底层指令实现其次是内存管理国产平台对动态shape的处理不如CUDA成熟很多地方要手动优化再就是通信库的调试多卡并行时DDP的实现细节有差别容易出现看起来每个设备都正常、但整个任务不收敛的情况。所以迁移前一定要给团队留足时间和心理预期。别指望一声令下两周就切换完成。4.2 纸面性能与实际吞吐的落差官方宣传的TFLOPS和实际业务跑出来的吞吐之间往往有巨大的鸿沟。举例来说某款推理卡官方INT8算力标得挺高但跑我的模型时因为框架里某些算子的实现走的是通用Fallback路径实际只有宣传值的50%左右。这不是厂商造假而是算子优化覆盖问题——芯片厂商优先优化头部开源模型的热点算子冷门算子很可能没来得及做深度优化。应对方法就是自建基准。我建议每个候选芯片都拿业务真实模型和线上真实请求数据跑基准最少要测一周的稳定性包括持续推理、并发压力、故障恢复。只看一次性benchmark选型上生产很容易翻车。4.3 驱动、框架、算子的暗坑与锁版本技巧国产芯片因为迭代快驱动和软件栈版本更新频繁伴随而来的是不少版本兼容性的暗坑。我踩过几次坑之后总结出一套锁版本的方法首先部署时记录完整的软件版本清单包括驱动、运行时、算子库、推理引擎、框架版本其次每次升级前先在测试环境跑完整回归尤其是推理精度对比和性能对比要一条条过最后升级后至少留一周的观察期确认没有性能回退再全量推广。总之生产环境不是追新版本的试验田稳定永远是第一位的。还有一个容易忽略的点国产芯片大多有自己的编译工具链算子层面的编译时间可能长达几十分钟这会拖慢日常开发迭代节奏。最好采用“增量编译算子缓存”的方式把常用的算子模板化能在你调试业务的每一天里省出大量时间。5. 常见问题与排查技巧实录5.1 模型跑不起来的常见原因很多人拿到国产卡后第一反应是“这也不行那也不行”但大部分问题其实是可归类的。最常见的是算子缺失、精度支持和框架版本不匹配。排查时先看算子日志确认缺失算子清单再决定是在算子库找替代还是调整模型结构规避。我建议第一次跑就直接上典型模型别拿玩具模型测——玩具模型跑通了不代表真实业务模型跑得通很多问题要在真实模型上才暴露。5.2 多卡扩展效率差怎么排查多卡训练效率差先区分问题是出在计算还是通信上。可以用一个简单的办法分别记录单卡纯算时间和多卡通信时间。如果通信占比超过30%八成是互联配置或者数据加载有瓶颈如果通信正常但加速比还是低那就要看负载均衡了有变量维度的单点热点都会导致其他卡空转。5.3 推理时延上不去的优化手段推理时延优化我总结的优先级先量化再算子融合最后才调批处理。量化是最直接的INT8/INT4对速度提升立竿见影前提是量化工具链靠谱。算子融合是把多个小算子合并成一个大算子减少内核启动和显存读写如果推理引擎支持自动融合优先打开。批处理要小心批处理越大吞吐越高但延迟也会涨所以要在延迟SLA和吞吐之间做平衡。我把常见的坑整理成了一张表现象可能原因排查方向模型转换报算子错算子库覆盖不全打印缺失算子清单评估手动实现成本推理结果精度不对量化误差或FP16精度问题对比FP32基线检查量化敏感层多卡训练不收敛通信库版本不一致或梯度聚合异常检查通信库版本开启同步调试日志显存不足崩溃动态shape导致显存预分配失误固定输入shape或用显存池优化性能与宣传差距大算子走了Fallback路径用profiler找热点算子反馈厂商优化6. 几点个人建议最后说点实在的。做了几年国产AI芯片的选型、迁移和上线工作我的感受是这个市场已经从“能不能用”进入“怎么用好”的阶段。算力芯片和推理专用芯片不是替代关系而是同一套AI基础设施里的不同齿轮。如果你的团队预算充足、以训练为主优先看生态成熟度最高的头部训练卡如果你的业务重心在大模型推理落地那国产推理专用芯片的性价比值得你专门立项评估。我个人最推荐的做法是不管最终买哪家一定要先做一轮真实模型验证把适配工作量、性能、成本、风险都量化出来。别被纸面参数冲昏头脑也别对国产品牌抱有成见——实际测一测很多刻板印象都会打破有些国产推理卡在同功耗下的性价比实测下来反而比进口卡更适合业务场景。国产AI综合生态正处在快速迭代的窗口期现在开始建立自己的选型方法论和评测基准后面无论市场怎么变你都能比大多数人从容。