在前面本地部署成功的基础上,按照如下命令执行:

text
CUDA_VISIBLE_DEVICES=0,1 \
torchrun --nproc_per_node=2 generate.py \
  --task ti2v-5B \
  --size 1280*704 \
  --ckpt_dir ./Wan2.2-TI2V-5B \
  --dit_fsdp \
  --t5_fsdp \
  --ulysses_size 2 \
  --prompt "一只白猫戴着墨镜坐在冲浪板上,夏日海边,电影感光影,镜头缓慢推进"

然后通过:

text
watch -n 1 nvidia-smi

观察:

text
Every 1.0s: nvidia-smi                                                                                                                                                                                                                                                                                                                       ai: Wed Apr 15 11:57:36 2026

Wed Apr 15 11:57:36 2026
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 590.48.01              Driver Version: 590.48.01      CUDA Version: 13.1     |
+-----------------------------------------+------------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA GeForce RTX 4090        Off |   00000000:01:00.0  On |                  Off |
|  0%   48C    P2            100W /  450W |   21616MiB /  24564MiB |    100%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
|   1  NVIDIA GeForce RTX 4090        Off |   00000000:05:00.0 Off |                  Off |
| 30%   40C    P2             86W /  450W |   21608MiB /  24564MiB |    100%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI              PID   Type   Process name                        GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|    0   N/A  N/A          114633      C   ...chenqi/venvs/wan22/bin/python      21598MiB |
|    1   N/A  N/A          114634      C   ...chenqi/venvs/wan22/bin/python      21598MiB |
+-----------------------------------------------------------------------------------------+