如何解决Jupyter Lab中ModuleNotFoundError: No module named 'tensorflow.keras'？

系统环境信息 #

Windows version :  11
Python version  :  3.10.7
Tensorflow      :  2.11.0
pip             :  22.3.1

已尝试但无效的方案 #

• 针对「ModuleNotFoundError: No module named 'tensorflow.keras'」的相关解决方案
• 针对「Traceback Error: ModuleNotFoundError: No module named 'tensorflow.keras'」的相关解决方案

报错场景 #

偏好使用virtual env而非Conda，通过非Conda方式安装TensorFlow，在Jupyter Lab中运行代码时触发错误：

运行以下代码：

import sys
sys.version

输出结果：

'3.10.7 (tags/v3.10.7:6cc6b13, Sep 5 2022, 14:08:36) [MSC v.1933 64 bit (AMD64)]'

接着运行代码：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf
import tensorflow.keras as keras   

触发错误：

ModuleNotFoundError Traceback (most recent call last)
Input In [2], in <cell line: 5>()
3 import matplotlib.pyplot as plt
4 import tensorflow as tf
----> 5 import tensorflow.keras as keras

ModuleNotFoundError: No module named 'tensorflow.keras'

解决方案 #

1. 对齐Jupyter Lab与TensorFlow的Python环境 #

Jupyter Lab可能默认使用了未安装TensorFlow的Python环境，需将其切换到你的virtual env：

• 在激活的virtual env中安装ipykernel：

  pip install ipykernel
  

• 将当前虚拟环境注册到Jupyter：

  python -m ipykernel install --user --name=你的虚拟环境名 --display-name="Python (你的虚拟环境名)"
  

• 重启Jupyter Lab，新建Notebook时选择刚才注册的环境。

2. 重新安装完整的TensorFlow #

若TensorFlow安装不完整导致keras模块缺失，执行以下操作：

• 卸载现有TensorFlow：

  pip uninstall tensorflow -y
  

• 清理缓存后重新安装指定版本：

  pip install --no-cache-dir tensorflow==2.11.0
  

3. 更换导入方式 #

如果上述方法无效，可尝试以下导入方式：

import tensorflow as tf
from tensorflow import keras
# 或单独安装keras后直接导入
# pip install keras==2.11.0
# import keras

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