周志洋

个人站

持续学习 才能不被淘汰


RTX PRO 6000 上起 vLLM:Blackwell 的坑比模型多

问题

PRO 6000(Blackwell,约 96GB)单卡可以扛 30B 级和较长上下文,FP8 吞吐也比 BF16 好。直接套 A100 镜像时,环境认不出这张卡,服务起不来。

环境问题

旧的 A100 推理镜像直接拿来用,典型失败:

  • PyTorch 看不到 sm_120
  • vLLM attention backend 选错,eager 模式慢得离谱
  • 驱动和 CUDA 用户态不一致

处理:这张卡单独一套环境,较新驱动 + CUDA 12.8/13 路线 + 对应 vLLM 构建。不要和 A100 池共用一个「万能」wheel。

冒烟命令先确认:

nvidia-smi
python -c "import torch; print(torch.cuda.get_device_name(0), torch.cuda.get_device_capability(0))"

算力显示 (12, 0) 一类再继续装服务。

单卡试跑策略

这张卡主要用来:

  • 验证某量化档(尤其 FP8)能不能起、质量抽检过不过
  • 试长上下文配置,再决定要不要迁到 A100 多卡
  • 给内部工具/编码助手做「单机专用」推理,不进公共高峰池

公开材料里 FP8 相对 BF16 常见有 2~3 倍量级的吞吐提升,同模型同长度抽测,方向一致。默认优先 FP8,除非业务强要求全精度。

# 示意:具体模型名随当时仓库而定
vllm serve <model-fp8> \
  --host 127.0.0.1 --port 8001 \
  --max-model-len 32768 \
  --gpu-memory-utilization 0.90 \
  --kv-cache-dtype fp8_e4m3

绑定 127.0.0.1 再经网关反代,避免 Docker 把端口直接曝到公网。

和 A100 的差异(体感)

  • 互联:单卡没事;若将来两张 PRO 只有 PCIe,TP 时可能要关 P2P/自定义 all-reduce,和 NVLink A100 不是同一套手感
  • 基线不同:同一模型在 PRO 6000 与 A100 上的 TTFT 不能共用一个告警阈值
  • 定位:试验床 + 专人工作负载,不是 8 卡池的第九张「同构卡」

这张卡跑通之后,新模型先在这里冒烟,再决定占几张 A100。省下的是主池被试验配置打挂的次数。


转载请注明:周志洋的博客 » RTX PRO 6000 上起 vLLM:Blackwell 的坑比模型多

打赏一个呗

取消

感谢您的支持,我会继续努力的!

扫码支持
扫码支持
扫码打赏,你说多少就多少

打开支付宝扫一扫,即可进行扫码打赏哦