CUDA error: an illegal memory access was encountered

编程知识 更新时间:2023-04-06 11:03:56

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

更多推荐

CUDA error: an illegal memory access was encountered

本文发布于:2023-04-06 11:03:00,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/8c8fa6326e0e9974bda81ade9e4571aa.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:illegal   error   CUDA   encountered   access

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!

  • 49202文章数
  • 14阅读数
  • 0评论数