限制skflow中的GPU内存分配

Question

我正在使用Tensorflow和skflow来训练凸网，在一个与其他人共享的EC2实例上。为了让我们能够同时工作，我想限制分配的可用GPU内存的一部分。

这个问题使用Tensorflow来实现它，但是由于我使用的是sklfow，所以我从不使用tf.Session()。

有没有可能通过滑雪来做同样的事情呢？

Answer 1

此时，您只能通过将此参数传递给估计器来控制用于估计器中的核心数(num_cores)。

可以将gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)添加到tf.ConfigProto中，如您所链接的这个问题所建议的那样，以达到您所需的目的。

请随意提交一个PR来进行更改，这里以及将这个附加参数添加到所有估计器中。否则，我会在这个星期的某个时间做些改变。

编辑：

我已经作出了修改，以允许这些选择。请检查示例文件夹中的“使用不同GPU配置构建模型”示例。如果你有什么特别的需要或其他选择，请告诉我。拉请求总是欢迎！