目前的代码只能在GPU:0中运行吗？

[sukasu403]

pred_depth, confidence, output_dict = model.inference({'input': rgb})
我将模型和上述代码中的rgb通过to(device)移到GPU:1后会在代码的其他部分报错：
Metric3D/mono/model/decode_heads/RAFTDepthNormalDPTDecoder5.py", line 215, in compute_depth_expectation
depth = torch.sum(prob * depth_values, 1)
~~~~~^~~~~~~~~~~~~~
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0!
也就是说，目前代码中的数据都是全部默认放置在GPU:0中的？