KD35lrp 2024-12-02 17:09 采纳率: 0%
浏览 10
已结题

不同GPU数据传输很神奇的现象

data1 = torch.randn(1, 3, device=d2l.try_gpu(0))
data2 = torch.randn(1, 3, device=d2l.try_gpu(1))
# 打印原始张量的内容和设备
print("data[0]:", data1)
print("data[0] device:", data1.device)
print("data[1]:", data2)
print("data[1] device:", data2.device)

# 使用 .to() 方法并打印
data1_to_device0 = data2.to(data1.device)
print("data[1] after .to(data[0].device):", data1_to_device0)
print("device of the new tensor:", data1_to_device0.device)

代码如上,在运行过程中data1_to_device0 = data2.to(data1.device)这一句在to其他device中居然把数值给修改了,这是为什么啊哪里出错了,改成了device0中的数据了。运行结果如下:

img

  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 12月10日
    • 创建了问题 12月2日