Gewu_Wang

我也遇到了同样的问题，请问你是怎么解决的呀？

libtorch-gpu代码中，没有显式的释放显存。在调用量增加的时候，是否会存在out of memory的问题？