Memory TensorFlow:不允许contrib.learn中的GPU内存增长选项。估计员
当使用contrib.learn创建估计器时,我能找到的唯一GPU选项是“GPU内存分数”,它可以在“tf.contrib.learn.RunConfig”中找到 在使用contrib.learn时,是否有方法允许GPU内存增长选项Memory TensorFlow:不允许contrib.learn中的GPU内存增长选项。估计员,memory,tensorflow,gpu,config,Memory,Tensorflow,Gpu,Config,当使用contrib.learn创建估计器时,我能找到的唯一GPU选项是“GPU内存分数”,它可以在“tf.contrib.learn.RunConfig”中找到 在使用contrib.learn时,是否有方法允许GPU内存增长选项 config = tf.ConfigProto() config.gpu_options.allow_growth = True config.gpu_options.per_process_gpu_memory_fraction = 0.9 run_config
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
config.gpu_options.per_process_gpu_memory_fraction = 0.9
run_config = tf.contrib.learn.RunConfig(session_config=config)
将上述配置传递给会话\ u配置
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
config.gpu_options.per_process_gpu_memory_fraction = 0.9
run_config = tf.contrib.learn.RunConfig(session_config=config)