Python中Tensorflow包安装使用存在的问题

这篇博客记录我使用TensorFlow包存在的问题。

问题一:导入时报错ModuleNotFoundError: No module named '_pywrap_tensorflow_internal'

场景:在我上一篇博客中提到了离线迁移python环境中的依赖包,其中有TensorFlow1.9.0.(CPU版本)

环境:WIN10 Python3.6

参考网上,可以检查下VS2015环境,我安装了一个VS2015的环境就解决了。

可能还有其他原因,可以参考以下链接:

https://blog.csdn.net/u010099080/article/details/53418159

https://github.com/tensorflow/tensorflow/issues/17393

问题二: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'
忽略掉警告。

参考这篇文章,https://blog.csdn.net/hq86937375/article/details/79696023,可以把警告消除。

第二种方法

卸载现有TensorFlow,安装支持cpu用AVX2编译的TensorFlow。

pip uninstall tensorflow

这里https://github.com/fo40225/tensorflow-windows-wheel下载支持的版本。

我的是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

有问题的话可以反馈给我,一起解决啦~~~

猜你喜欢

转载自blog.csdn.net/chengyikang20/article/details/81060819