0
点赞
收藏
分享

微信扫一扫

[深度学习][原创]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

就不会报错了

举报

相关推荐

0 条评论