TensorFlow能否创建多个会话？如何统计当前会话数量？

1. 是否允许创建多个会话？ #

当然可以！TensorFlow完全支持创建多个独立的tf.Session实例。每个会话都拥有自己的计算资源（比如GPU/CPU内存、变量状态），它们之间相互隔离——你在一个会话里初始化的变量，不会影响另一个会话中的同名变量。这种特性在需要同时运行多个独立的计算任务时非常有用。

2. 如何统计当前存在的会话数量？ #

TensorFlow并没有提供原生的API来直接获取当前活跃的会话数量，但我们可以通过猴子补丁的方式，对tf.Session的构造和销毁方法进行包装，从而维护一个会话计数器。下面是具体的实现方案：

完整示例代码 #

import tensorflow as tf

# 初始化会话计数器
_session_count = 0

# 对Session的构造方法进行包装，创建会话时增加计数
original_init = tf.Session.__init__
def patched_init(self, *args, **kwargs):
    global _session_count
    _session_count += 1
    original_init(self, *args, **kwargs)

tf.Session.__init__ = patched_init

# 对Session的close方法进行包装，手动关闭会话时减少计数
original_close = tf.Session.close
def patched_close(self, *args, **kwargs):
    global _session_count
    _session_count -= 1
    original_close(self, *args, **kwargs)

tf.Session.close = patched_close

# 处理with语句自动关闭会话的情况，确保计数正确更新
original_exit = tf.Session.__exit__
def patched_exit(self, exc_type, exc_val, exc_tb):
    global _session_count
    # 只有会话未被手动关闭时，才减少计数
    if not self._closed:
        _session_count -= 1
    return original_exit(self, exc_type, exc_val, exc_tb)

tf.Session.__exit__ = patched_exit

# 实现会话数量统计打印函数
def print_number_of_sessions():
    print(f"当前活跃的会话数量：{_session_count}")

# 测试函数
def test_print_number_of_sessions():
    sess1 = tf.Session()
    print_number_of_sessions()  # 输出：当前活跃的会话数量：1
    sess2 = tf.Session()
    print_number_of_sessions()  # 输出：当前活跃的会话数量：2
    
    sess1.close()
    print_number_of_sessions()  # 输出：当前活跃的会话数量：1
    
    # 使用with语句创建会话，自动关闭后计数会减少
    with tf.Session() as sess3:
        print_number_of_sessions()  # 输出：当前活跃的会话数量：2
    print_number_of_sessions()  # 输出：当前活跃的会话数量：1

# 运行测试
test_print_number_of_sessions()

代码说明 #

• 我们通过替换tf.Session的__init__方法，在每次创建会话时自动增加计数器；
• 替换close方法，确保手动关闭会话时计数器同步减少；
• 替换__exit__方法，覆盖了with语句自动关闭会话的场景，避免计数遗漏；
• print_number_of_sessions函数直接读取全局计数器的值并打印，实现需求中的统计功能。

需要注意的是，这种方式依赖于TensorFlow的内部实现细节，如果后续版本对Session类的方法签名或内部属性（比如_closed）进行修改，可能需要微调代码，但在大多数稳定版本中，这个方案都能正常工作。

内容的提问来源于stack exchange，提问作者mrgloom