内存溢出 pycharm pytorch Buy new RAM

not enough memory: you tried to allocate 655565 bytes.

错误如图所示， 最近经常报这样的错误…

原因找到了 自己定义的储存loss的list 储存的不是数值，是tensor…
导致每一个epoch结束后，内存一直在增加。

修改前：

修改后：

这样list存的就是float数值，而不是一个tensor！！！！！！！！！

另外转载一篇博客用来检查内存变化。

断点测试每一个epoch的内存变化。图为cpu代码。 博客里有gpu的。