pytorch 报错:RuntimeError: CUDA error: invalid configuration argument

    科技2026-01-26  12

    问题: RuntimeError: CUDA error: invalid configuration argument

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

    Processed: 0.032, SQL: 9