Как конвертировать ndarray в переменную autograd в формате GPU?

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 .