comfyUI报错-视频生成-, cuda:0 and cpu

Question

我在使用comfyUI的时候发生了报错 “”Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat1 in method wrapper_CUDA_addmm) 是一个视频生成流程时候出现的，运用了ComfyUI-FramePackWrapper节点

Answer 1

阿里嘎多学长整理AIGC生成，因移动端显示问题导致当前答案未能完全显示，请使用PC端查看更加详细的解答过程

问题解答

您遇到的报错是comfyUI在生成视频时出现的错误，错误信息是“Expected all tensors to be on the same device, but found at least two devices”，这意味着comfyUI期望所有张量都在同一个设备上，但是却发现至少两个设备。

这个错误可能是由于您使用了多个设备（cuda:0 和 cpu），而comfyUI不支持在多个设备上运行。comfyUI是一个基于GPU的视频生成工具，它需要所有的计算操作都在同一个GPU上进行。

解决这个问题的方法是：

1. 确保所有的张量都在同一个设备上，可以使用torch.device来指定设备。
2. 如果您需要使用多个设备，可以尝试使用torch.cuda.device_count()来获取当前可用的GPU数量，然后使用torch.cuda.set_device()来指定设备。

例如：

import torch
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
torch.cuda.set_device(device)

这将设置当前设备为cuda:0，如果cuda:0不可用，则设置为cpu。

如果您仍然遇到问题，可以尝试使用comfyUI的其他参数来控制设备的使用，例如：

comfyUI(..., device="cuda:0")

这将强制comfyUI使用cuda:0设备。

如果您需要更多帮助，请提供更多的代码和错误信息，我将尽力帮助您解决问题。