[深度学习][原创]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最先出现在Python成神之路

版权声明:
作者:ht
链接:https://www.techfm.club/p/16323.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>