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

CUDA error: an illegal memory access was encountered相关推荐

RuntimeError: CUDA error: an illegal memory access was encountered
文章目录 4. 我的解决 1. 错误描述 2. 自我尝试 2.1 减小batch_size 2.2 换卡改代码 3. 调研情况 4. 我的解决后续发现其实是某张卡有问题, 0~3一共4个GPU,只在 ...
【解决汇总：待更新】CUDA error: an illegal memory access was encountered with channels_last
为方便查看,就直接贴图和网址了.仅供学习使用,如有侵权,请联系我删除. 一.调试方法: 链接原文: Recently, I encountered "an illegal memory a ...
再次遇到RuntimeError: CUDA error: an illegal memory access was encountered
之前遇到过一次记录下来了,第一次遇到翻看之前的记录,首先想着如何让pycharm准确地显示错误. 1:os.environ['CUDA_LAUNCH_BLOCKING'] = '1',模型前加这句, ...
RuntimeError: CUDA error: an illegal memory access was encountered的解决方法
在Transformer模型中遇到这个问题参考方法:https://www.jianshu.com/p/e1a0b14916f9 原因 class Add_Norm(nn.Module):def _ ...
CUDA RuntimeError: CUDA error: an illegal memory access was encountered
我的CUDA version=11.4 百度.谷歌上能找的办法都找了,有各种解决办法减小batch_size的把所有net和输入都放入到相同设备的 net的train和eval状态检查的防止内存 ...
torch.backends.cudnn.enabled = False会引起CUDA out of memory和CUDA error: an illegal memory access was
一般来说,题目所示这种问题都是由batch_size的设置引起的,修改batch_size之后就能有效解决.但是,我今天遇到一种情况,这个问题不是由batch_size引起的,而是由t ...
caffe 报错 Check failed: error == cudaSuccess (77 vs. 0) an illegal memory access was encountered
之前从没有遇到这样的问题,找了三个小时的bug和资料后,在此处发现了解决方案,他是这么说的: In my case the "top" and "bottom" ...
caffe 报错 Check failed: error == cudaSuccess (77 vs. 0) an illegal memory access was encounteredcaffe
caffe 报错 Check failed: error == cudaSuccess (77 vs. 0) an illegal memory access was encountered 训练时候 ...
在yolo训练的时候又去测试就会报错：cuda error: out of memory
在yolo训练的时候又去测试就会报错:cuda error: out of memory, cuda.c Assertion '0' failed. 不过,如果是用的yolo-tiny.cfg的话是 ...

CUDA error: an illegal memory access was encountered

CUDA error: an illegal memory access was encountered相关推荐

最新文章

热门文章