[深度学习][原创]detectron2报错RuntimeError: CUDA error: device-side assert triggered

花明

关注

阅读 12

2024-11-04


这个错误确实很难解决,我在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

就不会报错了

精彩评论(0)

0 0 举报