RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/

作者：花生_TL007 | 2024-02-16 01:03:20

踩

cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/

Pytorch中debug目标检测网络时出现的问题

pytorch实现目标检测网络时，可能会出现一下问题：
RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:16

出现上述错误可能是因为在box_utils.py脚本中 conf = labels[best_truth_idx] + 1，从而使得所有的label都加上了1。解决方法删除 +1就OK了。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/花生_TL007/article/detail/89149