场景:我是使用jupyter notebook运行pytorch在cuda上跑的时候遇到的问题。


完美解决-RuntimeError: CUDA error: device-side assert triggered : 这里提到说是数目类别的问题,比如四类分类,理论上标签one-hot最大值是3,出现了4就会报这个错误。

上面的本质说法和记录一个error–CUDA error: device-side assert triggered是一样的:就是一个检索越界的问题。

我的问题和前面的都不一样,而且cpu模式下能够运行,gpu就gg了。
最后尝试直接重启jupyter的kernel

真的不报错了

但是也没有任何运行的输出结果。
之前有错误还能百度,现在直接卡死,欲哭无泪。

就cpu跑吧。

慢点就慢点。

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐