tensorflow中使用tf.ConfigProto()配置Session运行参数

tf.ConfigProto()函数用在创建session的时候，用来对session进行参数配置：

config = tf.ConfigProto(allow_soft_placement=True, allow_soft_placement=True)  
config.gpu_options.per_process_gpu_memory_fraction = 0.4  #占用40%显存  
sess = tf.Session(config=config) 

一、动态申请显存

config = tf.ConfigProto()  
config.gpu_options.allow_growth = True  
session = tf.Session(config=config)

二、限制GPU使用率

config = tf.ConfigProto()  
config.gpu_options.per_process_gpu_memory_fraction = 0.4  #占用40%显存  
session = tf.Session(config=config) 

或者

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