TensorFlow Datasets模块导入失败，安装后仍报错求助

问题详情 #

• 通过Anaconda执行conda install -c anaconda tensorflow-datasets安装时，出现环境解析警告：

Solving environment: - warning libmamba Added empty dependency for problem type SOLVER_RULE_UPDATE
done.

• 当前环境版本信息：

  • python 3.10.14
  • tensorflow 2.16.1 (pypi_0 pypi)
  • tensorflow-datasets 4.9.4 (pypi_0 pypi)

• 尝试过pip安装，但问题未解决，在Jupyter中运行代码时触发以下错误：

ModuleNotFoundError                       Traceback (most recent call last)
Cell In[1], line 5
      2 import tensorflow as tf
      3 from tensorflow import keras
----> 5 import tensorflow_datasets as tfds
      6 from sklearn.metrics import confusion_matrix, accuracy_score 
      7 import matplotlib.pyplot as plt

ModuleNotFoundError: No module named 'tensorflow_datasets'

解决步骤 #

1. 确认Jupyter使用的环境是否正确 #

Jupyter常默认使用base环境，导致找不到其他环境安装的包：

• 在Jupyter单元格中运行以下代码，查看当前Python路径：

  import sys
  print(sys.executable)
  

• 对比你安装tensorflow-datasets的Anaconda环境路径（可通过conda activate 你的环境名后执行which python查看）。若路径不一致，按以下操作添加对应内核：

  # 激活目标环境
  conda activate 你的环境名
  # 安装ipykernel
  conda install ipykernel
  # 将环境添加到Jupyter内核列表
  python -m ipykernel install --user --name=你的环境名
  

  重启Jupyter后，在右上角选择新添加的内核运行代码。

2. 彻底重装tensorflow-datasets到目标环境 #

残留的安装文件可能导致识别异常：

# 激活目标环境
conda activate 你的环境名
# 卸载所有已安装的tensorflow-datasets版本
pip uninstall tensorflow-datasets -y
conda uninstall tensorflow-datasets -y
# 重新用conda安装（优先conda以保证依赖兼容）
conda install -c anaconda tensorflow-datasets

3. 检查版本兼容性 #

虽然当前tensorflow 2.16.1与tensorflow-datasets 4.9.4理论兼容，但仍可尝试调整版本：

• 升级tensorflow-datasets到最新版：

  pip install tensorflow-datasets --upgrade
  

• 若仍有问题，可降级tensorflow到2.15.x版本（该版本与更多tfds版本稳定兼容）：

  pip install tensorflow==2.15.0
  

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