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

ht • • 好物分享

这个错误确实很难解决，我在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
分享
二维码
< <上一篇
下一篇>>