Tensorflow 如何用tf.estimator控制GPU内存大小

我试图控制分配给一个tensorflow估计器tf.estimator.estimator的GPU内存大小。目的是只分配一半在同一GPU上运行其他tensorflow网络。我找到的是contrib版本，但不是官方版本。有人知道这是否可能？

当您创建一个

估计器

实例时，您可以传入构造函数的

配置

实例。

RunConfig

有一个

session\u config

属性，您可以使用该属性设置会话参数的

tf.ConfigProto

在代码中，这转换为：

session\u config=tf.ConfigProto（）
session\u config.gpu\u options.per\u进程\u gpu\u内存\u分数=0.5
估计器配置=tf.estimator.RunConfig（会话配置=session配置）
my_estimator=tf.estimator.estimator（…，config=estimator_config）