tensorflow設定視訊記憶體自適應和視訊記憶體比例

2021-07-26 13:17:08 字數 470 閱讀 5387

**:

用慣了theano.再用tensoflow發現一執行視訊記憶體就滿載了,嚇得我吃了乙個蘋果。 

用**搜尋引擎毛都搜不到,於是翻牆找了下問題的解決方法,原來有兩種 

1. 按比例

config = tf.configproto()

config.gpu_options.per_process_gpu_memory_fraction = 0.4

session = tf.session(config=config, ...)

按需求增長(theano那種)

config = tf.configproto()

config.gpu_options.allow_growth = true

session = tf.session(config=config, ...)

頂 0 踩

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

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

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

用慣了theano.再用tensoflow發現一執行視訊記憶體就滿載了,嚇得我吃了乙個蘋果。用 搜尋引擎毛都搜不到,於是fq找了下問題的解決方法,原來有兩種 config tf.configproto config.gpu options.per process gpu memory fractio...

tensorflow2 0視訊記憶體設定

遇到乙個問題 新買顯示卡視訊記憶體8g但是tensorflow執行的時候介面顯示只有約6.3g的視訊記憶體可用,如下圖 即限制了我的視訊記憶體,具體原因為什麼我也不知道,但原來的視訊記憶體小一些的顯示卡就沒有這個問題。目前的解決辦法是 官方文件解決 然後對應的中文部落格 總結一下,就是下面的兩個辦法...