这篇博客记录我使用TensorFlow包存在的问题。
问题一:导入时报错ModuleNotFoundError: No module named '_pywrap_tensorflow_internal'
场景:在我上一篇博客中提到了离线迁移python环境中的依赖包,其中有TensorFlow1.9.0.(CPU版本)
环境:WIN10 Python3.6
参考网上,可以检查下VS2015环境,我安装了一个VS2015的环境就解决了。
可能还有其他原因,可以参考以下链接:
问题二:import tensorflow没问题,但是警告Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2
原因:我也只是理解了个大概就是CPU支持AVX2但是TF不支持。具体讨论可以看点击打开链接
环境:WIN10 Python3.6
第一种方法
在代码中加入
import os
os.environ["TF_CPP_MIN_LOG_LEVEL"]='2'
忽略掉警告。
参考这篇文章,可以把警告消除。
第二种方法
卸载现有TensorFlow,安装支持cpu用AVX2编译的TensorFlow。
pip uninstall tensorflow
在这里下载支持的版本。
我的是WIN10 Python3.6 CPU版本支持AVX2,就下载下图所示whl包,安装即可。
pip install tensorflow-1.9.0-cp36-cp36m-win_amd64.whl
参考文章https://blog.csdn.net/Fourierrr_/article/details/79749899