[SOLVED] Make Sure That Pytorch Using GPU To Compute

[pinocchio (Rene Sandoval)]

is that equivalent to:

    use_cuda = torch.cuda.is_available()
    device = torch.device("cuda" if use_cuda else "cpu")
   model = MyModel()
   model.to(device)

?

What is the difference between doing `net.cuda()` vs `net.to(device)`?

I was going through this post ([SOLVED] Make Sure That Pytorch Using GPU To Compute) and I had the question, what is the difference between these two pieces of code? import torch.nn as nn net = nn.Sequential(OrderedDict( [ ('fc1',nn.Linear(3,1)) ]) ) net.cuda() vs import torch import torch.nn as nn use_cuda = torch.cuda.is_available() device = torch.device("cuda" if use_cuda else "cpu") net = nn.Sequential( OrderedDict([ ('fc1', nn.Linear(3,1)) ]) ) net.to(device) vs import torch import t…