为啥训练完成后显存没有释放

[kc-w]

[AllentDan]

torch 的显存最好手动释放吧

[kc-w]

我用线程跑的训练,训练结束后内存和显存都没有释放,在网上也查了下用cudaDeviceReset()可以释放,但是调用过再次训练就没法加载模型了,像torch.cuda.empty_cache()这样的函数,libtroch里也没有,作者有什么好办法吗?

[AllentDan]

用进程吧，torch 的显存不容易释放干净