使用TensorFlow导入LSTM模型时出现导入错误

Hey there! Let's work through this issue together—since you're new to TensorFlow and Python, this is a common fixable problem, so don't stress.

首先，先聊聊那个GPU警告：

2021-08-19 10:42:59.820385: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2021-08-19 10:42:59.821236: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

这条警告完全不用在意！它只是提示你的电脑没有配置NVIDIA GPU，TensorFlow会自动切换到CPU模式运行，不会影响你使用LSTM或其他核心功能。

核心问题：ImportError: cannot import name 'ShardingPolicy' #

这个错误通常是由于TensorFlow版本不兼容或者安装文件损坏导致的，下面是几个可行的解决步骤：

• 步骤1：卸载并重新安装稳定版TensorFlow
  首先卸载当前可能有问题的TensorFlow版本：

  pip uninstall tensorflow -y
  

  然后安装一个与Python 3.9兼容的稳定版本（推荐LTS长期支持版2.15.0，对Python3.9支持非常友好）：

  pip install tensorflow==2.15.0
  

• 步骤2：确认版本兼容性
  确保你的Python版本（3.9）和TensorFlow版本匹配：TensorFlow 2.5到2.15版本都支持Python3.9，尽量避免安装太新或太旧的版本，以免出现不兼容问题。

• 步骤3：清理pip缓存后重新安装（如果步骤1无效）
  如果卸载重装后问题依然存在，可能是pip缓存的损坏文件在搞鬼，先清理缓存：

  pip cache purge
  

  然后再重新安装TensorFlow：

  pip install tensorflow==2.15.0
  

验证是否解决问题 #

打开Python终端（或你的Pycharm项目），依次运行以下代码：

import tensorflow as tf
from tensorflow.keras.layers import LSTM

如果没有报错，说明问题已经解决啦！

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