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 
      ) 
    ) 

有沒有一種方法可以讓GPU使用contrib.learn時的內存增長選項？

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