cuda error invalid argument

用pytorch训练时报的错

在loss.backward时报错

估计应该是loss出现了异常值。

网上其他答案：

问题：
RuntimeError: CUDA error: invalid configuration argument

原因：参数过大，内存需求过大
解决方法：查看之前的代码及网络设计是否合理，是否有错误的地方。本人错误在于之前忘记了加池化层，导致数据过大，从而报错。

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/118157168