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

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

This question 使用 Tensorflow 完成，但由于我使用的是 sklfow，因此我从不使用 tf.Session() 。

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

最佳答案

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

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

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

编辑:

我已经进行了更改以允许这些选项。请检查示例文件夹中的“使用不同的 GPU 配置构建模型”示例。如果您有任何特殊需要或要添加的其他选项，请告诉我。拉请求总是受欢迎的!

关于tensorflow - 限制 skflow 中的 GPU 内存分配，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/35483367/