Как мне узнать, хранится ли экземпляр на GPU с Pytorch?

Question

Я недавно изучаю Pytorch, и возникает этот вопрос. Например, если у меня есть net, унаследовавший «torch.nn.Module».

class Net(torch.nn.Module):
    def __init__(self, something):
        super(net, self).__init__()
        self.p1=something

    def forward():
        pass

net1=Net(123)
net1.cuda()  ##Here I can't see what is changed.

Тогда как я могу узнать, хранится ли net1 (и что-то такое) в GPU.

Я читал, как работает * .cuda (), похоже, что все "дети" запускают * .cuda (). Я пытался увидеть, что такое "дети". Похоже, что у net1 нет детей.

Answer 1

Чтобы проверить простой тензор, вы можете проверить атрибут is_cuda. Например:

x = torch.zeros(100).cuda()
y = torch.zeros(100)

print(x.is_cuda) # True
print(y.is_cuda) # False

Чтобы проверить модель, подумайте, что проще всего использовать метод parameters(), который возвращает все обучаемые параметры вашей модели.

next(model.parameters()).is_cuda

Cheers