Windows10安装Keras与TensorFlow时遇_pywrap_tensorflow模块缺失错误
解决TensorFlow
_pywrap_tensorflow 模块缺失问题 嘿,这个错误我之前帮不少人排查过,大概率是版本不兼容或者安装包与你的环境不匹配导致的——你装的是TensorFlow 0.12.0的CPU版本,但同时配置了CUDA 9和cuDNN 7,而且这个版本实在太老了,和CUDA 9完全不兼容(TensorFlow 0.12.0只支持到CUDA 8)。下面给你几个一步步解决的方案:
1. 先彻底卸载旧版本TensorFlow
先把这个不兼容的老版本清掉,避免后续冲突:
python3 -m pip uninstall tensorflow
2. 安装和CUDA 9、cuDNN 7匹配的TensorFlow版本
CUDA 9 + cuDNN 7对应的稳定TensorFlow版本是1.12.0~1.14.0,其中1.14.0的兼容性最好:
- 如果是Linux/Windows系统,直接装GPU版本:
python3 -m pip install tensorflow-gpu==1.14.0
- 如果是Mac系统,注意TensorFlow从1.13开始就不再支持Mac的GPU加速了,那你得装对应版本的CPU版:
python3 -m pip install tensorflow==1.12.0
3. 验证安装是否成功
装完之后,运行这段测试代码确认:
import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello))
如果能正常输出b'Hello, TensorFlow!',就说明没问题了。
额外排查小要点
- 确认你的CUDA和cuDNN环境变量配置正确:比如
CUDA_HOME、LD_LIBRARY_PATH(Linux)或者DYLD_LIBRARY_PATH(Mac)是否指向了正确的安装路径 - 检查Python版本是否匹配:TensorFlow 1.14.0支持Python 3.5~3.7,别用太新的Python版本
- 如果是Windows系统,还要确认装了对应版本的Visual Studio编译环境(比如VS2017,这是CUDA 9的要求)
内容的提问来源于stack exchange,提问作者Darlyn
相关产品推荐
相关产品推荐

