1. 程式人生 > >Tensorflow設定視訊記憶體自適應,視訊記憶體比例

Tensorflow設定視訊記憶體自適應,視訊記憶體比例

用慣了theano.再用tensoflow發現一執行視訊記憶體就滿載了,嚇得我吃了一個蘋果。
用天朝搜尋引擎毛都搜不到,於是翻牆找了下問題的解決方法,原來有兩種
1. 按比例

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4
session = tf.Session(config=config, ...)
  1. 按需求增長(theano那種)
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...
)