CUDA error: an illegal memory access was encountered

报错原因1:
这种报错的是模型model、输入数据(input_image、input_label)没有全部移动到GPU(cuda)上,或者没有移动到同一块gpu上。
**温馨提示:**debug时一定要仔细检查是否每一个输入变量以及网络模型都移动到了GPU上,我一般报错都是因为漏掉了其中一两个。

解决方法:
将model、input_image、input_label全部移动到cuda上,实例代码如下:

方法1:

model = model.cuda()
input_image = input_iamge.cuda()
input_label = input_label.cuda()

方法2:

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
model = model.to(device)
input_image = input_iamge.to(device)
input_label = input_label.to(device)

方法3:

model = model.cuda(2)
input_image = input_iamge.cuda(2)
input_label = input_label.cuda(2)

情况2:

我遇到的一种特殊情况是,只有cuda(0)不报错&

你可能感兴趣的:(pytorch知识宝典)