Wan2.2 中文文档 下载 App

显存需求一览表:从 RTX 4090 到 A100 各跑什么

核心结论先看

如果你只有一张消费级显卡(24GB 及以下),只能跑 TI2V-5B。A14B 系列(T2V-A14B / I2V-A14B)和 S2V-14B、Animate-14B 都需要至少 80GB 显存,适合 A100 / H100 或多卡 FSDP。

模型显存要求可用硬件
TI2V-5B≥ 24 GBRTX 4090(24GB)、3090(24GB)
A14B 系列(T2V / I2V)≥ 80 GBA100 80GB、H100 80GB、或多卡 FSDP
S2V-14B≥ 80 GB同上
Animate-14B≥ 80 GB同上

单卡 vs 多卡

单卡模式下,A14B 系列使用 --offload_model True --convert_model_dtype 参数把非活跃层卸载到 CPU,降低峰值显存,但速度会慢很多。如果有多张 GPU,可以用 FSDP + DeepSpeed Ulysses 做张量并行,速度提升明显。

官方仓库里附带了多卡启动命令模板,核心参数是:

不同 GPU 上的实际效率

官方在 GitHub 上发布了各模型在不同 GPU 上的测试结果,格式是 总时间(秒)/ 峰值显存(GB):

GPU 效率图
不同 GPU 上的计算效率对比

降低显存的常用技巧

跑 OOM(Out-of-Memory)时报错,可以依次尝试以下选项:

  1. --offload_model True:把模型层卸载到 CPU,速度变慢但显存降低
  2. --convert_model_dtype:把模型参数从 float32 转为 bfloat16,显存减半
  3. --t5_cpu:把 T5 文本编码器放在 CPU 上跑,牺牲速度换显存
  4. 降低分辨率:A14B 系列支持 480P 和 720P,先用 480P 验证
  5. 减少视频时长:默认 5 秒,改成 2 秒也能省下一些显存

这几个参数组合起来,可以把 A14B 的单卡显存需求从 80GB 降到 60GB 左右,但速度会明显变慢(大约 2~3 倍时间)。

消费级显卡的唯一选择

TI2V-5B 是消费级显卡用户的首选。它只需要 24GB 显存,在 RTX 4090 上生成 5 秒 720P 视频约 9 分钟。更多细节和调优技巧见 4090 本地跑。