[深度学习][原创]detectron2报错RuntimeError: CUDA error: device-side assert triggered
Posted FL1623863129
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了[深度学习][原创]detectron2报错RuntimeError: CUDA error: device-side assert triggered相关的知识，希望对你有一定的参考价值。
这个错误确实很难解决，我在GTX1060显卡下训练完全没问题，但是到了RTX3090显卡开始报错RuntimeError: CUDA error: device-side assert triggered，一模一样代码，结果报错。
GTX1060显卡的torch版本
torch==1.8.1+cu102
但是RTX3090如果torch仍为1.8.1就不行了，经过测试只要
torch==1.9.0+cu111
就不会报错了
以上是关于[深度学习][原创]detectron2报错RuntimeError: CUDA error: device-side assert triggered的主要内容，如果未能解决你的问题，请参考以下文章