tensorflow设置显存自适应和显存比例.

转自:http://blog.csdn.net/cq361106306/article/details/52950081

用惯了theano.再用tensoflow发现一运行显存就满载了，吓得我吃了一个苹果。 
用天朝搜索引擎毛都搜不到，于是翻墙找了下问题的解决方法，原来有两种 
1. 按比例

config = tf.ConfigProto()config.gpu_options.per_process_gpu_memory_fraction = 0.4session = tf.Session(config=config, ...)
1
2
3
1
2
3

1. 按需求增长(theano那种)

config = tf.ConfigProto()config.gpu_options.allow_growth = Truesession = tf.Session(config=config, ...)
1
2
3
1
2
3

顶

0

踩