导入tensorflow.keras遇ModuleNotFoundError：找不到对应模块求助

问题场景 #

运行以下代码时触发错误：

import numpy as np

from tqdm.auto import trange, tqdm
import matplotlib.pyplot as plt

import tensorflow as tf

from tensorflow.keras import layers

错误信息：

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[14], line 8
      4 import matplotlib.pyplot as plt
      6 import tensorflow as tf
----> 8 from tensorflow.keras import layers

ModuleNotFoundError: No module named 'tensorflow.keras'

解决方案 #

• 检查并升级TensorFlow版本
  TensorFlow 1.x版本中，Keras是独立包而非TensorFlow的子模块。如果你的环境装的是TF1.x，要么升级到TF2.x（推荐），要么单独安装Keras包：

  • 升级TF：pip install --upgrade tensorflow
  • 安装独立Keras：pip install keras

• 验证TensorFlow安装有效性
  先执行以下代码确认TF是否正常安装：

  import tensorflow as tf
  print(tf.__version__)
  

  如果这一步报错，说明TF未正确安装，先卸载再重装：

  • 卸载：pip uninstall tensorflow
  • 重装：pip install tensorflow（GPU版本用pip install tensorflow-gpu，需匹配对应CUDA、cuDNN版本）

• 确认Python环境一致性
  如果你使用虚拟环境，确保激活了正确的环境后再安装和运行代码。比如conda环境需先执行conda activate your_env_name。

• 临时替换导入语句
  若暂时无法升级TF，可修改导入语句为：

  from keras import layers
  

  前提是已单独安装Keras包。

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