RuntimeError: CUDA error: device-side assert triggered

在服务器上跑SSD eval.py代码时,出错:
RuntimeError: CUDA error: device-side assert triggered
在网上查了很久自己也做了很多测试,大概将定位到出错代码处,但具体的错误仍然不知道,然后我又在本地的pycharm上进行了测试,报错:
Expected object of scalar type unsigned char but got scalar type bool for argument ‘other’`

出错处源代码如下:

suppress = torch.max(suppress,
		     (overlap[box] > max_overlap)

发现是(overlap[box] > max_overlap)的类型错误,改正后的代码如下:

image_boxes.append(class_decoded_locs[1 - suppress])
image_labels.append(torch.LongTensor((1 - suppress).sum().item() * [c]).to(device))
image_scores.append(class_scores[1 - suppress])

运行成功!
有些时候CUDA报错,不一定其本身的问题,可能是代码的问题,还是要仔细一点查看代码才行。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值