Tensorflow 限制skflow中的GPU内存分配

我正在使用Tensorflow和skflow对ConvNet进行培训，这是我与其他人共享的一个EC2实例。为了让我们能够同时工作，我想限制分配的可用GPU内存的比例

使用Tensorflow实现，但是因为我使用的是sklfow，所以我从来没有使用

tf.Session（）

---

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

此时，您只能通过将此参数传递给估计器来控制要在估计器中使用的核数（

num_cores

）

您可以按照链接的建议将

gpu\u options=tf.gpu选项（每个进程\u gpu\u内存\u分数=0.333）

添加到

tf.ConfigProto

，以实现所需

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

编辑：

---

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

我已经添加了选项。在答案中查看我的编辑。