[FIXED] How to check if a tensor is on cuda or send it to cuda in Pytorch?

Issue

I have a tensor

t = torch.zeros((4, 5, 6))

How to check if it is on gpu or not, and send it to gpu and back?

Solution

From the pytorch forum

use 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

When passing to and from gpu and cpu, new arrays are allocated on the relevant device.

Answered By - Gulzar