如何解决Jupyter Notebook中导入TensorFlow库的AttributeError问题

问题说明 #

执行import tensorflow as tf时触发以下错误：

AttributeError: module 'tensorflow._api.v2.compat.v2.__internal__' has no attribute 'register_load_context_function'

核心原因是TensorFlow和Keras版本不兼容——当前环境中单独安装的Keras版本，与TensorFlow自带的Keras组件版本不匹配，导致调用内部函数时找不到对应属性。

解决方案（分步骤操作） #

1. 卸载冲突的现有包 #

打开Anaconda Prompt（或你的终端），执行以下命令彻底卸载TensorFlow和单独安装的Keras：

pip uninstall -y tensorflow keras

注：-y参数会自动确认卸载，无需手动输入yes，简化操作步骤。

2. 重新安装适配版本的TensorFlow #

直接安装TensorFlow即可，它会自动包含完全适配的Keras版本，无需单独安装Keras：

• 安装最新稳定版：

pip install tensorflow

• （可选）指定稳定版本（若担心最新版有兼容性问题，可选择2.15.0这类经过验证的版本）：

pip install tensorflow==2.15.0

3. 验证安装结果 #

重启Jupyter Notebook，新建代码单元格执行以下代码：

import tensorflow as tf
print("TensorFlow版本：", tf.__version__)
print("Keras版本：", tf.keras.__version__)

如果能正常打印两个版本号，说明导入问题已解决。

关键注意事项 #

• 不要同时单独安装Keras和TensorFlow，TensorFlow官方已将Keras整合为内置组件，单独安装会导致版本冲突。
• 如果使用Anaconda虚拟环境，请确保先激活目标环境（执行conda activate 你的环境名称）后再执行上述命令，避免修改全局环境。

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