Commit fd49a083 by libei

add gpu_options to support high tf

parent 94469024
...@@ -315,6 +315,7 @@ def session_config(): ...@@ -315,6 +315,7 @@ def session_config():
gpu_options = tf.GPUOptions( gpu_options = tf.GPUOptions(
per_process_gpu_memory_fraction=FLAGS.worker_gpu_memory_fraction) per_process_gpu_memory_fraction=FLAGS.worker_gpu_memory_fraction)
gpu_options.allow_growth = True
config = tf.ConfigProto(allow_soft_placement=True, config = tf.ConfigProto(allow_soft_placement=True,
graph_options=graph_options, graph_options=graph_options,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论