在tf.contrib.learn.Estimator设置GPU利用率

由于本人有强迫症，不想让GPU用到100%。正所谓凡事留一线，日后好相见。。。

立马翻开官方文档

https://www.tensorflow.org/tutorials/using_gpu

官方代码。。

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4
session = tf.Session(config=config, ...)

But！！！

我用的是Estimator类，分装好了Session的调用，所以并不适用！！

那应该会有什么方法把参数传进去的吧。。

在tf官网的search框内输入tf.config

选中的框框内是可能的config类

再看看Estimator类的文档

简直如有神助！！

根据RunConfig的文档先码好

config_tf = tf.ConfigProto()
config_tf.gpu_options.per_process_gpu_memory_fraction = 0.85
config_estima = learn.RunConfig(session_config = config_tf

运行坐等效果！！

说没有这个参数！！神奇了我的TF！！

也许是我粗心打错了吧。。翻开文档好好看看。。。

__init__(
    master=None,
    num_cores=0,
    log_device_placement=False,
    gpu_memory_fraction=1,
    tf_random_seed=None,
    save_summary_steps=100,
    save_checkpoints_secs=_USE_DEFAULT,
    save_checkpoints_steps=None,
    keep_checkpoint_max=5,
    keep_checkpoint_every_n_hours=10000,
    evaluation_master='',
    model_dir=None,
    session_config=None
)

白纸黑字写的啊。。。

此处省略1万字和一万头草泥马。。。

解决方法

原来在编译器里真没有这个参数，但是找到了

gpu_menmory_fraction这个参数

最后用

config_estima = learn.RunConfig(gpu_menory_fraction = 0.8)

查看GPU使用率

搞定！！

是官网的跟新太慢还是什么原因呢。。。