显存使用控制

如果不设置，tensorflow会默认占用所有的GPU显存。解决方法如下，GPU会根据使用的情况根据需求增加GPU显存的使用

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