RuntimeError: CUDA error: device-side assert triggered

RuntimeError: CUDA error: device-side assert triggered

欢迎来到英杰社区，这里是博主英杰https://bbs.csdn.net/topics/617804998

 

原因：

cuda运行可能是异步的（asynchronously），因此报错信息中提示的位置可能不准确。

解决办法：

情况一、

       切成CPU运行，就会找到报错的位置

情况二：

        很多教程都提出此问题出现的根本原因：超出限制

        可以考虑换利用torch.clamp切到过大或者过小的值。 看看loss是否设计的正常。

1. 建议通过设置环境变量 CUDA_LAUNCH_BLOCKING=1 来解决报错位置不准确的问题。在代码中可以通过以下方式设置

import os
os.environ['CUDA_LAUNCH_BLOCKING'] = "1"

  这会强制 CUDA 以同步方式运行，从而确保报错信息能够准确指向问题所在。

该错误可能是因为代码中存在维度不匹配或索引溢出等问题。例如，scatter_ 函数中目标索引的维度与被映射对象的维度不匹配。

调试建议：

• 当遇到此类错误时，建议先通过设置 CUDA_LAUNCH_BLOCKING=1 确定具体错误位置。

• 然后检查报错位置的代码，重点关注是否：①、维度不匹配 ②、索引超出范围  ③数据类型不一致