如何释放gpu显存

模型训练完后保存模型,然后代码自动进入数据集的测试,提示内存不够,请问如何用代码释放占用的gpu或者如何释放模型训练过程中所使用的gpu显存?
已邀请:

要回复问题请先登录注册