清空 Lua / Torch 中的变量以释放内存（GPU 或 CPU）

2016-6-17 2:16:30
收藏：0
阅读：71
评论：1

我正在使用 Lua/Torch 进行深度学习任务。有时我需要加载非常大的模型，这会导致内存成本非常高，例如：

net = torch.load('./path/to/the/model.t7')
// 执行完该行代码后，GPU 内存成本将升至 2800MB 的 3064MB

在这种情况下，如果我想清空变量，就像在 Matlab 中使用“clear var;”命令一样，该怎么办？

点赞
用户5342251
用户5342251

你可以像下面这样为变量赋值为空值：

net = nil;

为了进行垃圾回收，你也可以运行：

collectgarbage();

此外，所有使用局部前缀声明的变量在定义它们的代码块完成后被销毁。

2016-06-17 07:45:33

作者:

用户1659534

用户1659534
推荐文章