在skflow中限制GPU内存分配

Question

我正在与Tensorflow和skflow一起训练，在我与其他人分享的EC2实例上。为了让我们所有人能够同时工作，我想限制分配的可用GPU内存的比例。

This question使用Tensorflow来做，但由于我使用了sklfow，我从不使用tf.Session()。

是否可以通过skflow做同样的事情？

Answer 1

此时，您只能通过将此参数传递给估算器来控制估算器中使用的核心数（num_cores）。

可以按this question建议将gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)添加到tf.ConfigProto以实现您的需要。

随意提交PR以进行更改here，并将此附加参数添加到所有估算工具中。否则，我将在本周的某个时间进行更改。

编辑：

我已做出更改以允许这些选项。请在示例文件夹中选中“使用不同GPU配置构建模型”示例。如果您想要添加任何特殊需求或其他选项，请告诉我。欢迎提出拉动请求！