清除变量以释放Lua / Torch中的内存(GPU或CPU)

时间:2016-06-17 02:16:30

标签: lua torch

我正在使用Lua / Torch来完成深度学习任务。有时我会加载一个非常大的模型,这会导致内存成本非常高,例如:

net = torch.load('./path/to/the/model.t7') 
// after this line, the GPU memory cost raises to 2800MB of 3064MB

在这种情况下,如果我想清除变量,就像在命令clear var;的Matlab中一样?

1 个答案:

答案 0 :(得分:4)

你可以像这样给你的变量分配一个零值:

net = nil;

对于垃圾收集,您还可以运行:

collectgarbage();

此外,使用本地前缀声明的所有变量在完成定义的代码块后都会被销毁。