Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! 怎么办

您可以使用`.to()`方法将所有张量移动到同一设备上。例如，如果您的张量在CPU和CUDA设备上分散，您可以使用以下代码将其移动到相同的设备上：

import torch

# 假设张量t1在CPU上，t2在CUDA上
t1 = torch.tensor([1, 2, 3])  # 默认在CPU上
t2 = torch.tensor([4, 5, 6]).cuda()  # 移动到CUDA设备上

# 检查每个张量的设备
print(t1.device)  # 输出：cpu
print(t2.device)  # 输出：cuda:0

# 将t1移动到CUDA设备上
t1 = t1.to('cuda')

# 再次检查每个张量的设备
print(t1.device)  # 输出：cuda:0
print(t2.device)  # 输出：cuda:0

在上述示例中，我们使用了`.to()`方法将`t1`张量移动到CUDA设备上，使得两个张量都在同一设备上进行计算。