When I try to put enitre TainingData/Labels to GPU, the run seems to abruptly end without any error at line op = net(x). But when I put individual inputs and labels to GPU the run works perfectly. Can anyone explain why ?
Also, pytorch tutorial mentions “Remember that you will have to send the inputs and targets at every step to the GPU too:”. Why not entire input/target at the beginning, instead of at every step ?
#TainingData = torch.Tensor(TainingData).to("cuda:0")
#Labels = torch.Tensor(Labels).to("cuda:0")
for epoch in range(1000):
    for i in range(1, TainingData.shape[0]+1):
        x, y = TainingData[i-1].reshape(1,1,28,28), Labels[i-1]
        y = torch.tensor([y], dtype=torch.long).to("cuda:0")
        x = torch.Tensor(x).to("cuda:0")
        op = net(x)
        loss = criterion(op, y)