方式一
tudui = Tudui()
if torch.cuda.is_available():
tudui = tudui.cuda (网络调用cuda后要再返回给网络)
..........................................................................................
if torch.cuda.is_available():
loss = loss.cuda()
..........................................................................................
if torch.cuda.is_available():
imgs = imgs.cuda()