更改TensorFlow会话的默认配置？

Question

是否可以在Python中或通过设置环境变量等来更改默认的Session配置？

具体来说我想

with tf.Session() as sess:
    ...

当我与其他作业并行运行小型测试时，耗尽更少的内存。所以我上面的行为与

相同

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.1)
config = tf.ConfigProto(gpu_options=gpu_options)
with tf.Session(config=config) as sess:
    ...

Answer 1

不要认为这是设置流程范围默认设置的方法，但这是我使用的模式。

def create_session():
  config = tf.ConfigProto(log_device_placement=True)
  config.gpu_options.per_process_gpu_memory_fraction=0.3 # don't hog all vRAM
  sess = tf.InteractiveSession("", config=config)
  return sess

sess=create_session()
a=tf.constant(1)
b=tf.constant(2)
sess.run([a+b])