在 Torch 机器学习框架中设置 GPU 使用率答案

【问题标题】：Set GPU usage in Torch Machine Learning Framework在 Torch 机器学习框架中设置 GPU 使用率
【发布时间】：2017-05-15 04:43:36
【问题描述】：

目前我正在使用 Theano 进行机器学习，现在我想试试 Torch。

在 Theano 中，有一个选项可以设置 GPU 内存使用的标志：

os.environ["THEANO_FLAGS"] = "mode=FAST_RUN, device=gpu,floatX=float32,lib.cnmem=0.9"

所以 Theano 使用 GPU 的设定容量，这里是 90%。然而，在 Torch 中，类似的网络使用了大约 30% 的 GPU 负载。

有什么方法可以像 Theano 一样在 Torch 中设置更高的 GPU 负载？

【问题讨论】：

标签： lua gpu torch

【解决方案1】：

Torch 将根据其标准分配器使用尽可能多的 GPU 内存。

torch 使用的内存量不需要像 Theano 中的示例那样预先指定。

【讨论】：