cuda out of memory
解决方法:http://www.cnblogs.com/Charlene-HRI/p/10234656.html
我的问题时在测试时候需要保存每次的输出中间向量,忘记转为 cpu().numpy(),
导致显存累积,越来越大,最终溢出
cuda out of memory
解决方法:http://www.cnblogs.com/Charlene-HRI/p/10234656.html
我的问题时在测试时候需要保存每次的输出中间向量,忘记转为 cpu().numpy(),
导致显存累积,越来越大,最终溢出