方法一：
分別對網絡模型，數據（輸入，標注），損失函數調用.cuda()
網絡模型：
if torch.cuda.is_available():net=net.cuda()數據（訓練和測試）：
if torch.cuda.is_available():imgs=imgs.cuda()targets=targets.cuda()損失函數：
if torch.cuda.is_available():loss_fn=loss_fn.cuda()方法二：
.to(device)
Device=torch.device("cpu")
Torch.device("cude:0")
定義訓練設備
device=torch.device("cpu")網絡模型
net=net.to(device)損失函數
loss_fn=loss_fn.to(device)數據
imgs=imgs.to(device)
targets=targets.to(device)
)
