解决cuda error:device-side assert triggered

一、问题描述

如题所示,在训练网络时报错cuda error:device-side assert triggered

二、解决方法

(1)torch.backends.cudnn.enable =True, torch.backends.cudnn.benchmark = True
(2)代码逻辑问题:
1.检查代码,看是否是训练时loss成为nan,可以更换词向量的拼接方式等。
2.如果是分类任务,可能是标签个数不对应,特别是还有index >= -sizes[i] && index < sizes[i] && “index out of bounds的提示时。
(3)出现device-side assert triggered的BUG不能只关注报错代码
(4)可以直接打开pytorch的源码:如发生binary_cross_entropy_out_cuda()断言,可能是训练到一个时间点时,变成nan无穷值,这样即使再输入sigmoid最后一层,最终的结果也不会是0到1之间,可以将BCELoss改为BCEWithLogitsLoss
(5)也可以直接先使用CPU跑代码,可以显示出直接用GPU跑代码没有显示的错误。

Reference

[1] RuntimeError: CUDA error: device-side assert triggered的解决
[2] 关于cuda error:device-side assert triggered的3种解决办法

  • 6
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 8
    评论
CUDA error: device-side assert triggered错误通常是由于在GPU设备上执行的CUDA代码中出现了断言错误导致的。断言是一种用于检查程序中的条件是否满足的机制,当条件不满足时,断言会触发错误并终止程序的执行。 要解决CUDA error: device-side assert triggered错误,可以按照以下步骤进行排查和修复: 1. 检查CUDA代码:首先,检查你的CUDA代码中是否存在断言语句。断言通常以`assert(condition)`的形式出现,其中`condition`是一个布尔表达式。确保断言语句的条件正确,并且在满足条件时不会触发断言错误。 2. 检查输入数据:断言错误有可能是由于输入数据不符合预期导致的。检查你的输入数据是否满足CUDA代码中对数据类型、大小、范围等方面的要求。确保输入数据的正确性。 3. 检查内存分配:断言错误还可能与内存分配有关。检查你的CUDA代码中是否存在内存分配错误,比如使用了无效的指针或者越界访问了数组。确保内存分配和释放的正确性。 4. 启用错误检查:在CUDA程序中,可以通过设置CUDA运行时环境变量来启用错误检查功能,以便更详细地了解错误的具体信息。可以将以下代码添加到你的CUDA程序中,以启用错误检查: ```cpp cudaDeviceSynchronize(); cudaError_t error = cudaGetLastError(); if (error != cudaSuccess) { printf("CUDA error: %s\n", cudaGetErrorString(error)); } ``` 这段代码会在CUDA程序中的关键位置进行同步,并检查是否存在错误。如果有错误发生,它会打印出错误信息,帮助你更好地定位问题。 5. 查阅文档和论坛:如果以上步骤无法解决问题,可以查阅CUDA官方文档、开发者论坛或者其他相关资源,寻找类似问题的解决方案或者向社区寻求帮助。 希望以上步骤能够帮助你解决CUDA error: device-side assert triggered错误。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

山顶夕景

小哥哥给我买个零食可好

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值