tensorflow限制显存

wangli

Python在用GPU跑模型的时候最好开多进程，因为很明显这种任务就是计算密集型的。

用进程池好管理，但是tensorflow默认情况会最大占用显存，尽管该任务并不需要这么多，因此我们可以设置显存的按需获取，这样程序就不会死掉了。

按比例预留：

tf_config = tensorflow.ConfigProto()  
tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 
# 分配50%  
session = tensorflow.Session(config=tf_config)

或者干脆自适应然后自动增长：

tf_config = tensorflow.ConfigProto()  
tf_config.gpu_options.allow_growth = True # 自适应 
session = tensorflow.Session(config=tf_config)