首先,在最上面放进去
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
其次,网络进入device
net.to(device)
最后,数据进入device
inputs, labels =
首先,在最上面放进去
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
其次,网络进入device
net.to(device)
最后,数据进入device
inputs, labels =