pytorch训练过程中释放显存

在python程序中需要显存暴增的问题,可以试试
在代码中插入

torch.cuda.empty_cache()

你可能感兴趣的:(pytorch)