python - 如何将 ndarray 转换为 GPU 格式的 autograd 变量？

Question

我正在尝试做这样的事情，

data = torch.autograd.Variable(torch.from_numpy(nd_array))

它属于类型Variable[torch.FloatTensor]，但我Variable[torch.cuda.FloatTensor]还需要我想这样做，pytorch version 0.3.0其中缺少一些方法，如to(device)或set_default_device

Answer 1

您可以使用cuda()张量的方法。

如果您想使用特定设备，您可以使用上下文管理器，例如

with torch.cuda.device(device_index):
    t = torch.FloatTensor(1.).cuda()

有关更具体的信息，请查看版本 0.3.0 的文档。