admin管理员组

文章数量:1794759

CUDA error: an illegal memory access was encountered

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)不报错&

本文标签： illegalerrorCUDAEncounteredAccess

版权声明：本文标题：CUDA error: an illegal memory access was encountered 内容由林淑君副主任自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.xiehuijuan.com/baike/1686959489a122367.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。