Tensorflow設置顯存自適應,顯存比例
阿新 • • 發佈:2018-04-03
memory table 適應 true allow growth nbsp pro OS
1. 按比例
1 config = tf.ConfigProto() 2 config.gpu_options.per_process_gpu_memory_fraction = 0.4 3 session = tf.Session(config=config, ...)
or
1 session_bundle_config.mutable_session_config() 2 ->mutable_gpu_options() 3 ->set_per_process_gpu_memory_fraction(0.4);
2.按需求增長
1 config = tf.ConfigProto()2 config.gpu_options.allow_growth = True 3 session = tf.Session(config=config, ...)
or
1 session_bundle_config.mutable_session_config() 2 ->mutable_gpu_options() 3 ->set_allow_growth(true);
Tensorflow設置顯存自適應,顯存比例