如何在TensorFlow 1.14中隐式启用TensorFlow v2行为？

针对你不想显式调用tf.enable_v2_behavior()、希望通过环境变量或其他隐式方式全局启用v2行为的需求，这里有几个可靠的方案：

1. 使用环境变量TF2_BEHAVIOR=1 #

这是最推荐的方案，完全契合你“全局隐式生效”的需求：

• 设置方式：在Docker镜像中通过ENV TF2_BEHAVIOR=1指令添加，或者在服务启动脚本中用export TF2_BEHAVIOR=1配置。
• 生效逻辑：TensorFlow 1.14已经支持这个环境变量，当它被设为1时，TensorFlow在初始化阶段会自动触发tf.enable_v2_behavior()的底层逻辑，不需要在业务代码里做任何显式调用。
• 核心优势：所有运行环境（测试、生产等）只要继承这个环境变量，就会统一使用v2行为，彻底杜绝“遗忘调用函数导致测试与生产行为不一致”的风险。

2. 利用Python的sitecustomize.py自动加载 #

如果环境变量方案有特殊限制，还可以借助Python的自动加载机制实现：

• 操作步骤：在Python的site-packages目录下创建sitecustomize.py文件，写入以下内容：

  import tensorflow as tf
  tf.enable_v2_behavior()
  

• 生效逻辑：Python解释器启动时会自动执行sitecustomize.py中的代码，因此任何导入TensorFlow的脚本都会先启用v2行为，完全不需要业务代码做额外处理。
• 注意事项：需要确保所有运行环境的Python都包含这个文件，在Docker镜像中可以提前将该文件复制到对应目录。

补充说明 #

你提到的“TensorFlow主__init__.py会自动调用该函数”确实是TensorFlow 2.0及以后的默认行为，但在1.14版本中默认还是启用v1行为，所以必须通过上述方式触发自动启用v2的逻辑。

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