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

Posted 2023-03-29

【中文标题】在 Torch 机器学习框架中设置 GPU 使用率

【英文标题】：Set GPU usage in Torch Machine Learning Framework

【发布时间】：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 负载？

【参考方案1】：

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

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

【讨论】：

啊，谢谢！这就解释了为什么内存使用量会随着网络大小自动增加。