可以直接model.to(device)
但对于dataloader里面的变量
如
for i, data in enumerate(trainloader):
# get the inputs
xt, yt = data #torch.Size([32, 1, 28, 28])
#print(xt.size(),yt.size())
xt=xt.to(device) #
yt=yt.to(device)
必须显式赋值才行