Tensorflow设置显存自适应,显存比例

来源:互联网 发布:一起走数据怎么作弊 编辑:程序博客网 时间:2024/04/28 19:13

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

config = tf.ConfigProto()config.gpu_options.per_process_gpu_memory_fraction = 0.4session = tf.Session(config=config, ...)
  1. 按需求增长(theano那种)
config = tf.ConfigProto()config.gpu_options.allow_growth = Truesession = tf.Session(config=config, ...)
1 1
原创粉丝点击