tensorflow - 限制 skflow 中的 GPU 内存分配

Question

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

这个问题是用 Tensorflow 做的，但是因为我使用的是 sklfow，所以我从不使用tf.Session().

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

Answer 1

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

可以按照您链接的这个问题的建议添加gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)以实现您所需要的。tf.ConfigProto

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

编辑：

我进行了更改以允许这些选项。请查看示例文件夹中的“使用不同的 GPU 配置构建模型”示例。让我知道您是否有任何特殊需要或要添加的其他选项。拉请求总是受欢迎的！