import torch
if torch.cuda.is_available():
device = torch.device('cuda')
else:
device = torch.device('cpu')
判断gpu是否可用
import torch
if torch.cuda.is_available():
device = torch.device('cuda')
else:
device = torch.device('cpu')
判断gpu是否可用