之前看到这个错误,找了半天。
检查下面几点:
1 模型是否放到了CUDA上 model = model.to(device) 或 model = model.cuda(device)
2 输入数据是否放到了CUDA上 data = data.to(device) 或 data = data .cuda(device)
3 模型内部新建的张量是否放到了CUDA上 p = torch.tensor([1]).cuda(device) 或 p = torch.tensor([1]).cuda(device)
当时我犯的错就是模型没有放上去