语义分割训练报UDA error: an illegal memory access was encountered

最新推荐文章于 2025-05-13 10:50:51 发布

原创最新推荐文章于 2025-05-13 10:50:51 发布 · 574 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#python

收录于

该错误通常是由于标签类别数量不匹配或者图像格式错误导致。如果GroundTruth(GT)的最大标签值超过代码中设定的类别数，或者GT图像存储为24位图但未转换为8位灰度图，都可能导致此CUDA错误。解决方法包括检查并确保代码中的类别设置与GT一致，以及正确处理图像格式。

报错信息如下：
terminate called after throwing an instance of ‘c10::Error’
what(): CUDA error: an illegal memory access was encountered
Exception raised from create_event_internal at /opt/conda/conda-bld/pytorch_1603729128610/work/c10/cuda/CUDACachingAllocator.cpp:687 (most recent call first):
frame #0: c10::Error::Error(c10::SourceLocation, std::string) + 0x42 (0x7f401a8438b2 in
（1）一般报这个错误是因为GT中的标签与程序中设置的类别数量对不上，比如说GT中最大标签数值为35，但是代码中设置为了小于35的类别数量，导致报这个错误。
（2）还有另外一种可能，就是GT图像存成了24位图，读取GT后没有转换为8位的灰度图，最后导致类别错误。