TensorFlow-GPU环境下Keras导入异常问题及解决

问题场景 #

在conda的TensorFlow-GPU环境中运行DCGAN代码（参考Saul Dobilas的博客）时，触发错误：

ModuleNotFoundError: No module named 'keras'

排查过程 #

• 首次在Python解释器中执行from tensorflow import keras，报错依旧
• 重启Python解释器后，执行import tensorflow.keras as keras成功，且获取到Keras版本号为2.6.0

根因分析 #

问题源于TensorFlow版本异常：
TensorFlow 2.x版本已将Keras整合为内置子模块，正常情况下from tensorflow import keras与import tensorflow.keras as keras两种导入方式均应生效。出现单种方式失效，说明环境中TensorFlow安装存在损坏、依赖缺失，导致模块导入路径异常。

修复步骤 #

1. 卸载当前TensorFlow-GPU版本：

pip uninstall tensorflow-gpu -y

2. 重新安装匹配Keras 2.6.0的TensorFlow-GPU版本：

pip install tensorflow-gpu==2.6.0

3. 验证修复效果：
   启动Python解释器，执行以下命令确认无报错：

import tensorflow.keras as keras
print(keras.__version__)  # 预期输出：2.6.0
from tensorflow import keras
print(keras.__version__)  # 预期输出：2.6.0

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