@ptrblck ,
One more questions:
# code 1:
device=torch.device('cuda')
net.to(device)

# code 2:
device=torch.device('cuda:0') # 'cuda:1'
net.to(device)

what is difference between ‘cuda’ and ‘cuda:0’ or ‘cuda:1’? what case should I use them in?