在前面本地部署成功的基础上,按照如下命令执行:
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 |
+-----------------------------------------------------------------------------------------+
评论
欢迎留下反馈,评论发布后会立即显示。