TensorFlow：在contrib.learn中不允许GPU内存增长选项。估计

Question

使用contrib.learn创建估算器时，我只能找到的GPU选项是“gpu_memory_fraction”，可以在“tf.contrib.learn.RunConfig”中找到

classifier = learn.Estimator(
     model_fn=cnn_model_fn,
     config=learn.RunConfig(
           gpu_memory_fraction=0.9
           )
     )

使用contrib.learn时，有没有办法允许GPU内存增长？

Answer 1

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
config.gpu_options.per_process_gpu_memory_fraction = 0.9
run_config = tf.contrib.learn.RunConfig(session_config=config)

将以上配置传递给Estimator's session_config