我有Tensort = torch.zeros((4, 5, 6))如何检查它是否在gpu上，并将其发送到gpu和返回？
t = torch.zeros((4, 5, 6))
mwngjboj1#
From the pytorch forum使用t.is_cuda、t.cuda()、t.cpu()
t.is_cuda
t.cuda()
t.cpu()
t = torch.randn(2,2) t.is_cuda # returns False t = torch.randn(2,2).cuda() t.is_cuda # returns True t = t.cpu() t.is_cuda # returns False
当传递到GPU和CPU以及从GPU和CPU传递时，在相关设备上分配新的数组。
hjzp0vay2#
@Gulzar只告诉你如何检查Tensor是在cpu上还是在gpu上，你可以通过下面的方法在GPU上计算Tensor：
t = torch.rand(5, 3) device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") t = t.to(device)
2条答案
按热度按时间mwngjboj1#
From the pytorch forum
使用
t.is_cuda、t.cuda()、t.cpu()当传递到GPU和CPU以及从GPU和CPU传递时，在相关设备上分配新的数组。
hjzp0vay2#
@Gulzar只告诉你如何检查Tensor是在cpu上还是在gpu上，你可以通过下面的方法在GPU上计算Tensor：