Lua 在Torch机器学习框架中设置GPU的使用_Lua_Gpu_Torch

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

lua

Lua 在Torch机器学习框架中设置GPU的使用,lua,gpu,torch,Lua,Gpu,Torch,目前我正在使用Theano进行机器学习，现在我想试试Torch 在Theano中，有一个为GPU内存使用设置标志的选项： os.environ[“THEANO\u FLAGS”]=“mode=FAST\u RUN，device=gpu，floatX=float32，lib.cnmem=0.9” 所以Theano使用GPU的设置容量，这里是90%。然而，在Torch中，类似的网络使用大约30%的GPU负载有没有办法在Torch中设置与Theano类似的更高GPU负载？Torch将根据其标准分配器

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

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

os.environ[“THEANO\u FLAGS”]=“mode=FAST\u RUN，device=gpu，floatX=float32，lib.cnmem=0.9”

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

有没有办法在Torch中设置与Theano类似的更高GPU负载？

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

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

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