法一:
device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model=model.to(device)
x=x.to(device)
y=y.to(device)
法二:
model=model.cuda()
x=x.cuda()
y=y.cuda()
法一:
device=torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model=model.to(device)
x=x.to(device)
y=y.to(device)
法二:
model=model.cuda()
x=x.cuda()
y=y.cuda()