16张B200 vs 8张AMD：大模型部署真正的瓶颈是显存

先说结论这个标题真正有价值的部分不是“AMD 打败了 NVIDIA”而是它把大模型本地部署的讨论焦点从“算力不够”拉回到了“显存不够”。对于一个 2.8T 参数级别的 MoE 模型决定能不能跑起来的第一因素从来不是峰值算力而是权重放不放得下、专家参数能不能在 GPU 之间轮转。16 张 B200 是一个可接受的基线8 张 AMD GPU 能装下同样是可行的方向——关键要看的是显存容量、显存带宽、量化精度以及互连拓扑这四件事而不是被数字对比带偏。这篇文章只做一件事把“16 张 B200 才能跑”和“8 张 AMD 就装下了”这两句话拆开讲清楚背后的大模型部署原理、硬件差异和工程落地路径。读者看完之后至少能回答三个问题为什么这类超大模型吃显存AMD 方案凭什么能把 GPU 数量砍半如果自己想在一台 8 卡机器上部署类似模型该从哪一步开始又会栽在哪些坑上。1. 从标题说起16张B200和8张AMD比的到底是什么先把这个对比拆开。B200 是 NVIDIA 面向大规模训练和推理推出的加速卡单卡 HBM3e 显存容量远高于上一代 H100显存带宽也大幅提升。AMD 这边的对应产品主要是 Instinct 系列比如 MI300X、MI325X它们的特点是大显存、大带宽并且价格上通常比同代 NVIDIA 旗舰更有竞争力。按最粗浅的算法16 张 B200 的总显存规模和 8 张 AMD 大显存卡可能接近。这意味着如果某个模型在 FP8 或 INT4 精度下的权重大小超过单卡显存、却又没超过 8 卡总显存那么用更少的 AMD 卡就有机会把全部权重放进显存直接省掉大量 CPU offload 和跨节点通信。但要小心一点显存容量只是最表层的对比。真实部署里影响模型能不能“装下”的因素排序大致是模型权重和 KV Cache 的总显存需求单卡显存上限显存带宽特别是专家并行的场景下需要频繁读取不同专家权重GPU 之间的互连带宽决定了张量并行时通信开销CUDA 生态或 ROCm 生态的成熟度单位显存成本和整机功耗。只看“8 张 AMD vs 16 张 B200”这个标题容易让人觉得 AMD 方案只是“便宜大碗”。但更准确的判断是这种对比本质上是显存容量与软件生态之间的置换。AMD 用更大容量的单卡把卡数降下来但代价是需要接受 ROCm 生态里不少组件仍在追赶 CUDA 的现实。所以如果这篇文章只给你一句“AMD 赢了”那是不负责任。我更愿意说这个标题是对大模型部署成本结构的一次重新估算它把显存容量提升到了与算力同等重要的位置。这个思路对于预算有限、又想做本地超大模型验证的团队来说非常值得认真研究。2. Kimi K3为什么一个2.8T参数的模型会吃掉这么多显存Kimi K3 是一个参数规模达到 2.8T2.8 万亿级别的模型。这个数字从普通人的视角看是抽象的但如果把它对应到显存就非常具体了。以最粗略的计算方式如果直接用 FP16 精度加载权重2.8T 参数乘以 2 字节大约需要 5.6TB 显存。即使降到 8bitINT8 / FP8也需要约 2.8TB 显存再降到 4bit也需要约 1.4TB 才能把权重完整放进显存。而单张主流数据中心 GPU 的显存通常在 80GB 到 192GB 之间少数能达到 256GB。单卡装不下就必须把权重切到多张卡上这就是“16 张 B200 才能跑”这类说法的来源。Kimi K3 完全展开来看加载过程的显存消耗还会更大原因在于推理阶段还需要保存 KV Cache长上下文场景下 KV Cache 会占掉非常可观的显存计算图、激活值、中间结果都会产生额外显存开销如果使用张量并行每张卡都需要持有模型分片不是简单的“总显存大于权重就行”如果是 MoE混合专家架构虽然专家参数很多但每个 token 实际激活的专家有限所以理论上可以用更聪明的调度方式来减少峰值显存负担。这里要专门说一下 MoE。Kimi K3 是一个典型的超大参数 MoE 模型。MoE 的核心思路是把一个大模型拆成多个“专家”子网络每个 token 只路由到其中少数专家。这样总参数量可以做到非常大但每次推理实际计算的参数量远小于总参数量。也就是说2.8T 是“总参数”每次推理可能只需要激活一小部分专家。这个特性对部署来说是一把双刃剑。好处是如果部署框架能判断哪些专家经常被用到就可以把热专家常驻显存冷专家按需加载从而在有限的显存里运行超大模型。坏处是MoE 模型随机路由时任何两张卡都可能需要加载不同的专家权重如果互连带宽不够加载专家的耗时就会超过计算耗时整体吞吐直接崩掉。所以“Kimi K3 2.8T 模型核心原理”落到部署层面真正要理解的是它不是单纯的大而是“又大又稀疏”。这种稀疏性给了大家用较少 GPU 装下它的希望但也对显存带宽和节点内互连提出了很高要求。3. “8张AMD就装下”这个说法成立需要满足什么条件从纯粹的总显存角度8 张大显存 AMD GPU 确实有可能装下一个量化后的 Kimi K3。但这个说法能否在真实环境中成立取决于下面四个条件缺一不可。3.1 量化精度必须足够激进如果不做量化2.8T 参数在 FP16 下需要超过 5TB 显存8 张卡按平均 192GB 算也只有 1.5TB 左右完全不够。要做到“8 张装下”权重精度至少要到 8bit更现实的可能是 4bit 级别。4bit 量化的代价是模型精度可能下降具体下降多少要看模型的鲁棒性、量化校准数据集的质量以及是否使用 GPTQ、AWQ、SmoothQuant 这类更先进的量化方法。在实际项目中不能简单地认为“能装下就等于能好好跑”量化之后必须做评测对比。3.2 依赖 MoE 的稀疏加载能力即使量化到 4bit2.8T 参数也需要约 1.4TB 显存8 张 192GB 的卡勉强接近临界真正跑起来还要留出 KV Cache 和激活值空间。所以光靠量化还不够还需要利用 MoE 的稀疏性让权重按需加载到显存。具体来说部署框架需要支持“专家并行”或“权重分层存储”把最常用的共享参数和热专家放在显存把冷专家放在 CPU 内存或 NVMe SSD 上推理时按需换入。这个机制在 vLLM、SGLang 等框架中有不同形态的实现但在 AMD 平台上的支持成熟度还需要单独验证。3.3 显存带宽和互连不能成为瓶颈Kimi K3 这类模型即使只激活少量专家读取权重仍然是一个极大的带宽需求。AMD 大显存卡的 HBM 带宽理论上很高但节点内多卡互连比如 AMD 的 Infinity Fabric和跨节点网络能否稳定支撑高并发专家加载是需要打问号的。如果 8 张 AMD 卡部署后单卡之间的互连带宽被一张网卡或一个 PCIe Switch 卡住那么实际吞吐可能低于 4 张 B200 的小规模集群。这也是我反复强调的显存容量决定模型“能不能放进去”互连带宽决定模型“跑得快不快”。3.4 软件栈必须能跑起来这是“8 张 AMD 就装下了”这个标题里最容易忽略的部分。B200 方案背后是成熟的 CUDA 生态几乎所有主流推理框架都把 CUDA 作为第一优先支持。AMD 这边走的 ROCm 路线虽然近几年进步明显但仍有不少场景需要手动打补丁、选择特定容器镜像、甚至自己编译算子。结论很明确“8 张 AMD 装下”在理论上可行在实际中有条件。如果你想复现这个方案不要先买卡先做两件事第一确定量化方案和推理框架是否支持 AMD第二在一台小规模 AMD 机器上跑通一个几千亿参数的 MoE 模型验证显存和带宽是否真的够用。标题可以制造热度工程上还是要靠数据说话。4. 本地部署的第一步弄清环境与硬件边界如果你看完前面的分析决定在自己的 AMD 服务器上尝试部署一个超大 MoE 模型那首先要做的是环境准备。这里的“环境”不只是一条安装命令而是一条完整的技术链路。4.1 操作系统与驱动在 AMD GPU 上跑大模型推荐 Linux尤其是 Ubuntu 20.04/22.04 LTS。ROCm 对 Ubuntu 的支持最完善很多容器镜像也默认基于 Ubuntu。安装 ROCm 驱动前建议先确认 GPU 型号在 ROCm 官方支持列表里。社区里经常有人拿着消费级 AMD 显卡去跑 ROCm但消费级卡的 ROCm 支持一直处于“能用但可能随时出问题”的状态。数据中心的 MI 系列相对稳妥。在 Linux 下安装驱动的通用思路是# 以 Ubuntu 为例先更新系统 sudo apt update sudo apt upgrade -y # 添加 ROCm 官方源版本号以官方仓库为准 wget https://repo.radeon.com/amdgpu-install/latest/ubuntu/jammy/amdgpu-install_6.x.x_all.deb sudo apt install -y ./amdgpu-install_*.deb # 安装 ROCm sudo amdgpu-install --usecaserocm,rocmdev安装完驱动后重启并验证rocm-smi这个命令会显示所有 AMD GPU 的温度、显存占用、利用率。如果输出为空说明驱动或权限配置有问题不能进入下一步。4.2 容器环境大模型部署强烈建议使用容器。原因很简单ROCm 的依赖版本非常敏感直接装在宿主机上很容易因为一次 apt upgrade 把驱动环境弄坏。容器可以把运行环境锁定在一个稳定的镜像里。Docker 访问 AMD GPU 的常见方式是# 安装 Docker 后加入 docker 用户组 sudo usermod -aG docker $USER # 运行 ROCm 容器验证 GPU 可见性 docker run --rm --device/dev/kfd --device/dev/dri --group-addvideo \ rocm/dev-ubuntu-22.04:latest rocm-smi如果这条命令能正确显示 GPU 信息说明 Docker 已经能访问 AMD GPU。后面所有推理框架都应该在这个容器环境里运行而不是直接装在宿主机上。4.3 WSL2 的特殊情况如果你用的是 Windows 开发机并且希望用 WSL2 跑 AMD GPU那就要额外检查 WSL 内核里的 GPU 驱动支持。GPU 驱动需要同时满足 Windows 侧和 WSL 侧的要求属于最容易出问题的场景。一个常见现象是 Windows 里能识别显卡但进入 WSL2 后rocm-smi什么都看不到。这时候优先检查 Windows 下的 AMD 驱动是否支持 WSL2再检查 WSL 内核版本是否过旧。消费级开发机和数据中心服务器是两种完全不同的部署环境。如果你只是想在本地看一下 Kimi K3 这类模型的推理效果建议先不要直接上 8 卡集群而是先用 1 到 2 张卡跑一个量化后的小尺寸 MoE 模型把“驱动 - 容器 - 推理框架 - 模型加载”这条链路跑通。链路通了再扩展卡数。5. 从 Ollama 到 vLLM不同层级的部署方案对比部署超大 MoE 模型可选的路很多。对开发者和运维来说重要的是理解不同工具的使用边界。工具定位适合场景AMD 支持情况Ollama本地一键式推理快速体验、轻量调用有 AMD 方向支持但超大模型需谨慎vLLM高吞吐推理服务生产环境、高并发社区支持较多需要验证 ROCm 版本SGLang高性能推理框架复杂调度、多模型场景在 AMD 上逐步跟进Hugging Face Transformers研究与原型快速验证模型结构通用性能不是最优如果你的目标是“在自己电脑上跑起来”Ollama 可能是最容易入门的选项。它抽象了模型下载、量化、加载和推理的过程一条命令就能启动一个 OpenAI 风格 API。但 Ollama 的设计目标是个人开发者和轻量场景对多卡并行、专家并行、自定义量化策略的支持相对有限。2.8T 参数的模型即使能装下用 Ollama 管理也不一定是最佳选择。vLLM 则是更接近生产环境的方案。它支持 PagedAttention、Continuous Batching 等推理优化技术在多卡并行方面有更细粒度的控制。如果你的目标不是“跑起来看一眼”而是“作为服务对外提供服务”那么 vLLM 更合适。这里要强调一个原则不要因为搜索到一个“ollama for amd installer”就觉得万事大吉。先弄清你手里的模型、显存、驱动版本是否和框架版本匹配否则后面排查问题会非常痛苦。6. 用 vLLM 部署 MoE 大模型的最小示例在 AMD GPU 上 vLLM 已经支持 ROCm但具体到 Kimi K3 这种 2.8T 参数模型官方镜像未必直接支持所有量化格式。下面给出的是一个通用流程演示“如何拉起一个量化后的大模型服务”而不是某一个特定模型的承诺。6.1 拉取 vLLM 的 ROCm 镜像建议直接使用官方提供的 ROCm 版本镜像避免自己手动编译算子docker pull vllm/vllm-openai:latest如果你本机已经有编译好的 ROCm 环境也可以基于源码安装git clone https://github.com/vllm-project/vllm.git cd vllm pip install -e .6.2 启动 openai 兼容服务假设你已经准备好了一个量化后的模型目录默认存放在/models/kimi-k3-awq。启动服务的命令大致如下docker run --rm \ --ipchost \ --device/dev/kfd \ --device/dev/dri \ --group-addvideo \ -v /models:/models \ -p 8000:8000 \ vllm/vllm-openai:latest \ --model /models/kimi-k3-awq \ --tensor-parallel-size 8 \ --gpu-memory-utilization 0.9 \ --max-model-len 8192 \ --dtype float16几个参数简单解释--tensor-parallel-size 8表示将模型切分到 8 张 GPU 上并行推理--gpu-memory-utilization 0.9允许模型最多占用单卡 90% 的显存--max-model-len 8192限制最大上下文长度值越大KV Cache 占用越高--dtype float16加载权重时的精度如果你已经做了 AWQ/GPTQ 量化这里可能需要改成对应配置。这一步是“看起来简单实际上最容易报错”的地方。如果框架版本与 ROCm 版本不兼容或者模型量化格式不被支持服务会在加载阶段直接崩溃甚至还没有输出任何错误日志。6.3 调用模型服务服务启动后可以用 curl 验证curl http://localhost:8000/v1/chat/completions \ -H Content-Type: application/json \ -d { model: /models/kimi-k3-awq, messages: [{role: user, content: 请用一句话解释 MoE 模型}], max_tokens: 100 }如果返回 JSON 结果说明推理链路已经通了。如果长时间没有响应优先看服务日志而不是反复重试请求。6.4 用 Ollama 做一个更轻量的对比如果你更想先用 Ollama 快速演示可以参考下面的流程ollama pull 模型名 ollama run 模型名Ollama 本身会自动处理模型下载、量化格式转换和加载。它最大的价值是降低实验门槛但面对 2.8T 参数的模型它是否支持高效的专家并行和权重分层需要以官方文档为准。在一个还没验证过的新模型上不要把它当作生产环境服务来用。7. 如何判断部署真的成功了运行结果与效果验证很多人以为“服务起来了”就等于部署成功。实际上一个可靠的大模型部署至少要经过四个层面的验证。7.1 基础可用性验证服务启动后先请求一个短问题确认模型能正常生成。用 curl 就可以完成这一步重点看回复有没有乱码、是否完全中断、是否长时间无响应。7.2 显存与性能指标验证用rocm-smi查看每张 GPU 的显存占用。如果 8 张卡里有几张显存占用特别高、几张几乎空闲说明并行切分不合理模型推理性能会被拖累。watch -n 1 rocm-smi正常情况下8 张卡在推理时应该都有明显的显存占用和计算利用率。如果某张卡利用率始终为 0%大概率是并行策略没有生效。7.3 长上下文验证大模型部署最常见的隐性失败是短问题能回答长上下文下显存溢出。这是因为 KV Cache 随序列长度线性增长。建议用不同的输入长度做测试记录显存占用变化# 设置一个较长的 system prompt 或重复文本测试模型在长输入下是否稳定如果长文本场景下频繁 OOM就需要调低max-model-len或者优化 KV Cache 的量化策略。7.4 正确性对比部署一个量化后的超大模型精度下降是必然的。你不能只看“能生成”还要和未量化版本或官方 API 做一组评测样本的对比。具体做法是准备一组包含数学、代码、逻辑推理、长文总结的测试题分别用量化模型和基准模型跑一遍对比输出的一致性和质量。这一步不做你可能会把“模型变笨了”误判成“显存不够、部署有问题”浪费大量排查时间。8. 常见问题与排查思路结合 AMD GPU 上跑大模型的常见情况下面整理了一张排查表。问题现象可能原因排查方式解决方案rocm-smi看不到 GPU驱动未正确安装或权限不足检查dmesg确认 GPU 是否被内核识别重新安装 ROCm 驱动确认当前用户加入了video组容器无法访问 GPU未传递--device/dev/kfd --device/dev/dri或缺少--group-addvideo在容器内执行rocm-smi补全 Docker 参数后重试模型加载阶段 OOM权重精度过高、上下文过长查看服务启动日志统计显存占用降低量化精度调小max-model-len调低gpu-memory-utilization推理速度极慢显存带宽不足专家权重频繁从存储加载观察每张 GPU 利用率、CPU 负载和磁盘 IO使用更高带宽的卡优化模型并行策略把冷专家权重放到 NVMe SSD输出乱码量化精度严重下降或 tokenizer 配置错误对比官方模型的输出重新进行量化校准检查 tokenizer 文件是否匹配WSL2 环境识别不了 AMD GPUWindows 驱动不支持 WSL2 或内核过旧查看 Windows 驱动版本、WSL 内核版本更新 AMD 驱动和 WSL 内核这里特别提醒一句遇到问题先看日志不要盲目重装驱动。大多数看似“驱动坏了”的问题实际是权限配置、容器参数或内核模块加载顺序引起的。9. 最佳实践与工程建议9.1 从“能跑”到“能稳定运行”需要工程化设计如果你只是做技术验证直接把模型放进容器跑通即可。但如果你想把它作为团队内部的推理服务建议至少做到下面几点用版本管理工具锁定镜像版本、模型文件和量化配置为模型服务单独建一套监控记录显存占用、请求延迟、错误率预留温备 GPU 或 CPU offload 能力避免单卡故障导致全部服务不可用服务启动前做一次模型加载自检失败时自动告警而不是静默重启。9.2 量化不是“压文件”要参与精度评估很多人对 4bit 量化的理解是“把模型压缩一下省点显存”。这个想法很危险。量化会改变模型权重分布尤其对代码生成、数学推理这类对精度敏感的任务影响更大。实际项目中建议做一个标准评测集量化前后各跑一遍把差值记录下来作为是否接受该量化方案的依据。9.3 先做小规模验证再上 8 卡即使你的目标就是 8 卡 AMD 部署也不要在第一轮就买齐 8 张卡。先在 1 卡或 2 卡环境把推理框架、量化格式、并行策略选型确定下来再逐步扩展到目标卡数。这样可以大幅降低试错成本也更容易定位是软件问题还是硬件卡数问题。9.4 AMD 平台上的兼容性要保持敬畏ROCm 的进步是真实的但它在算子覆盖、第三方库适配、社区资料沉淀上和 CUDA 仍有差距。不要在 8 卡 AMD 环境下安装一个最新版本的框架就默认它一定能跑通 2.8T 模型。更稳妥的做法是先查官方支持矩阵再看社区里是否有同类硬件的成功案例最后在小规模环境下验证关键算子。9.5 不要只盯着 GPU存储和内存同样重要超大 MoE 模型在冷专家换入时实际上是从 CPU 内存或 SSD 读取权重。如果服务器只有 HDD或者 CPU 内存容量小于模型总大小部署必然失败。一个现实的配置建议是CPU 内存至少是模型总大小的 2 倍NVMe SSD 预留足够空间并优先选择顺序读取性能好的硬件。10. 总结与后续学习方向“16 张 B200 才能跑的 Kimi K38 张 AMD 就装下了”这个标题背后真正值得记住的知识点有三个。第一大模型部署的瓶颈正在从“算力不够”转移到“显存容量和带宽不够”。2.8T 参数的 MoE 模型能够被 8 张 AMD 卡装下核心原因是量化、专家稀疏加载和大显存容量三者叠加而不是某一项技术单点突破。第二AMD 方案的低 GPU 数量优势是真实存在的但它的代价是更高的工程成本。ROCm 生态、推理框架兼容性、长稳运行的稳定性都需要在选型时提前验证。第三无论用哪家硬件部署思路是通用的先量化评估再小规模验证并行策略最后再扩展到完整集群。不要因为一个吸引眼球的标题就直接跳进 8 卡集群的部署。如果你接下来想继续深入建议按这样的顺序学习先搞懂 MoE 和 KV Cache 的显存计算模型再熟悉一种推理框架的并行策略配置然后找一台 1 卡 AMD 机器跑通端到端推理链路最后再研究多机多卡场景下的通信优化和异常恢复。8 张 AMD 到底能不能稳定服务 Kimi K3这个问题的最终答案不在标题里而在你的评测报告里。