Cosmos 3 能跑世界模拟:部署前先算清三笔账

2026 年 11 月 2 日,NVIDIA 发布《Build World Models for Interactive Simulation with NVIDIA Cosmos》。这次更新不只是再添一个视频模型,而是把世界模型的训练、仿真、评估、优化与推理服务串成一套开发流程。现在值得看,是因为 Cosmos 3 Nano 已提供可运行检查点、仿真配方和部署入口,普通开发者不必先自行拼装完整研究栈。

世界模型可以先理解为“预测下一步世界会怎样变化”的模型:输入文字、图像、视频或动作,输出未来场景。在机器人、自动驾驶和手术策略评估中,它还能同步生成摄像头画面与激光雷达数据。下文凡标注“推算”的,都是我依据 NVIDIA 公布的配置和结果作出的判断,不是官方结论;其余产品能力与数字均来自官方资料。

从生成视频走向可验证场景

世界模型过去最容易被演示成一个“很会拍视频”的模型:输入一句话,得到一段看起来合理的道路或机器人画面。NVIDIA 这次公布的 Cosmos Framework 更关注一项更实际的能力——生成结果能否进入仿真流程,并接受自动检查。

以自动驾驶场景为例,开发者可以提供 World Scenario Map(WSM,世界场景图)控制项,用来描述车道、物体、道路标线、交通标志和道路边界。联合 RGB–LiDAR 生成还会使用高清地图的 rangemap 数据,并支持设置初始激光雷达观测。文字提示则补充天气、光照、时段和道路状况。

这意味着任务从“生成一段视频”变成“构造一个可控场景”。前者适合展示,后者才可能用于闭环测试。我的边界推导是:若不同传感器输出不能保持空间和时间一致,即使单看视频质量很高,也不宜直接拿它训练或验收下游策略。

三类配方对应三种真实风险

Cosmos Framework 提供的仿真配方覆盖机器人、真实场景生成、多摄像头驾驶仿真和手术策略评估。手术场景的专用检查点属 Cosmos3-H 系列——如 Cosmos3-H-Embodied-Nano(在 Open-H-Embodiment 上后训练,可从机器人动作生成手术视频、也可从视频预测动作)与它蒸馏出的 Cosmos3-H-Dreams(测试手术策略如何随进程演进)。通用的 Cosmos 3 Nano 检查点则支持 PyTorch 与 vLLM-Omni,官方称已在 NVIDIA Hopper 和 Blackwell GPU 上测试。

使用场景 输入控制 主要验证目标 常见误区
自动驾驶仿真 提示词、WSM、高清地图 rangemap 多车视角、雷达数据与道路约束是否一致 画面逼真就代表可用于闭环评估
机器人仿真 自定义传感器、动作空间和数据域 生成状态能否支持策略迭代 直接沿用通用检查点,忽略传感器布局
手术策略评估 场景与动作条件 关键观察是否保持连续、可追溯 只检查最终视频,不检查中间状态

自动驾驶与机器人配方最明显的差别,是可控变量的类型。前者大量依赖道路结构、天气和交通参与者,后者更强调动作接口、相机配置与实体数据域。NVIDIA 还提到,Cosmos Framework 正从 Cosmos 3 扩展到 Alpamayo、GR00T、Qwen、pi0.5 及其他兼容的 Hugging Face 模型。

真正的门槛在显存和延迟

根据官方表格,Cosmos 3 Nano 生成任务的显存分三档:BF16 需 58 GiB、FP8 需 44 GiB、FP8 并启用卸载只需 31 GiB(官方正文另有一处把 44 写成“FP8 with offloading”,与表格口径不一致——以表格三档为准)。BF16 是常见的低精度数值格式,保留接近常规浮点数的表达范围,同时减少计算和存储开销;FP8 则进一步压缩数值精度。

因此,BF16 与 FP8 不是“更高质量”和“更省显存”两个抽象标签,而是两套资源取舍:前者适合优先保留生成质量的配置,后者用于缓解显存压力。NVIDIA 称,Cosmos NIM 微服务通过量化、推测解码和感知 KV 缓存的路由,在 H200 GPU 上可获得约 2 至 3 倍加速。这里是官方给出的特定硬件结果,不能直接外推到所有 GPU,也不能据此承诺端到端业务同样提速。

一个可复核的参考运行如下,命令来自 NVIDIA 公布的 Transfer-Auto 示例:

OSMOS_TRAINING=0 COSMOS_INTERNAL=0 hf download nvidia/Cosmos3-Nano-Transfer-Auto 
  --local-dir checkpoints/Cosmos3-Nano-Transfer-Auto
AUTO_SAMPLE_ROOT="$PWD/checkpoints/Cosmos3-Nano-Transfer-Auto/assets/sample_data"
AUTO_NEGATIVE_PROMPT="$(cat .agents/skills/cosmos3-nano-transfer-auto/inference/assets/negative-prompt.txt)"

torchrun --standalone --nproc-per-node=4 
  -m cosmos_framework.scripts.inference 
  -i "$AUTO_SAMPLE_ROOT/input-transfer-auto.jsonl" 
  -o outputs/transfer_auto 
  --checkpoint-path checkpoints/Cosmos3-Nano-Transfer-Auto 
  --resolution 720 --num-steps 35 
  --per-view-negative-prompt "$AUTO_NEGATIVE_PROMPT" 
  --cp-size 4 --cfgp-size 1 
  --no-use-torch-compile --no-diffusion-cache

该参考运行会生成 7 段 720p 视频,每段 297 帧、帧率为 30 FPS,同时产生 99 次激光雷达扫描。这里的 7、297 和 99 是官方示例的输出规模,不是固定吞吐量指标。换句话说,这组数字适合用来估检查文件大小和验证工作量,不适合直接预测另一张地图或另一组相机的运行时间。

先检查数据,再相信画面

NVIDIA 对参考结果给出了具体检查项:查看 sample_outputs.json 是否标记成功,检查 lidar.safetensors 中的距离和强度是否为有限值,再核对 WSM 对齐情况,以及不同摄像头视图与激光雷达之间的一致性。

这类检查比“抽一帧看起来不错”可靠得多。生成式视频很容易出现道路结构漂移、物体位置冲突或时间连续性中断;激光雷达文件即使成功写出,也不代表每一束测量都合理。官方产品页同时列出 Cosmos Curator 和 Cosmos Evaluator,分别用于筛选、整理传感器数据,以及自动检查和评分生成视频,说明数据治理与结果评估本来就是同一套工作流的一部分。

这是我根据两个官方页面交叉看到的产品侧重点:技术博客强调检查点、命令、显存和就绪验证;Cosmos 产品页则把 Cosmos 3 放进更大的物理 AI 工具链中,涵盖视觉语言推理、世界行为模型后训练、数据整理和合成视频。官方文章不是独立使用一个模型,而是在补齐“生成之后如何进入工程流程”这一段。

适合先做什么,哪些情况别急

如果你的目标是原型验证、传感器适配、场景生成,或者需要比较不同策略在同一组条件下的表现,Cosmos Framework 的模块化设计值得尝试。训练侧提供 LoRA、全参数监督微调,以及 DPO、GRPO、PPO、OPD、DiffusionNFT 等在线或离线强化学习路径;推理侧可接 Hugging Face Diffusers、PyTorch、vLLM、TRT-LLM 和 NVIDIA NIM 等组件。

但如果目标只是生成一段宣传视频,完整引入 Cosmos Framework 可能过重。它更适合已有仿真、机器人或自动驾驶数据流程的团队:这类用户已经知道要约束什么,也能定义跨视角、跨传感器与时间连续性的验收标准。

按官方 NIM 容器化部署方式,开发者需要根据 GPU 选择配置、启动服务、检查就绪状态,再发送首个推理请求。具体选型应回到 NVIDIA NIM 文档与下载入口核对当前支持矩阵,而不是只看单次演示的加速数字。

真正该带走的是验收清单

Cosmos 3 的意义不在于把“世界模型”包装成一个更响亮的名称,而在于把生成、检查点和部署之间的接口摆了出来。对普通 AI 技术用户,判断一次世界模型演示是否可用,可以沿着四层检查:场景条件是否可控,输出是否同步,异常能否自动发现,服务是否适合目标 GPU。

目前官方资料同时给出了 Hopper、Blackwell 测试信息,也给出了明确显存与加速结果,但不同模型、精度、分辨率和卸载方式之间仍需单独验证。这是我根据官方部署信息作出的边界推导:Cosmos Framework 已经提供可运行的工程入口,但生成质量、物理一致性和业务吞吐量仍是三套不同验收题,不能用一段好看的视频替代。

参考来源

评论区

0 条评论

登录后可评论。

白鹿 74 阅读