要将CUDA上的变量移动到CPU上,请使用.cpu()
方法。例如:
import torch # 假设 x 在 GPU 上 x = torch.tensor(2.5, device='cuda') # 将 x 从 GPU 移动到 CPU x_cpu = x.cpu() # 现在可以在 CPU 上使用 x_cpu 变量 print(x_cpu)
在这个示例中,我们首先将x
张量对象创建在GPU上。然后,我们使用.cpu()
方法将其移动到CPU上,并将其分配给一个新的变量x_cpu
。现在,我们可以在CPU上使用x_cpu
变量并打印它。
请注意,将张量移动到不同的设备(如从GPU到CPU)可能会涉及到数据的复制,因此需要确保不会频繁地在不同的设备之间移动数据以避免性能下降。