3.1 第一种
找到:
1网络模型 .cuda()
2数据(输入,标注)(从数据加载器取出的数据).cuda()
3损失函数 .cuda()
3.2 第二种
.to(device)
device=torch.device(‘cpu”)/Torch.device(‘cuda’)/Torch.devich(‘cuda:0’)
3.1 第一种
找到:
1网络模型 .cuda()
2数据(输入,标注)(从数据加载器取出的数据).cuda()
3损失函数 .cuda()
3.2 第二种
.to(device)
device=torch.device(‘cpu”)/Torch.device(‘cuda’)/Torch.devich(‘cuda:0’)