【已解决】使用TensorFlow报错:“ModuleNotFoundError: No module named ‘tensorflow.contrib’”

问题描述:

运行基于tensorflow的代码,原代码在tensorflow v1的基础上编写,当前tensorflow大多是v2,因此运行时会出现下列错误:


原有方案

按照博客,需做如下修改:
将import tensorflow as tf 改为

  import tensorflow.compat.v1 as tf

然而,修改之后仍会继续报错:

ModuleNotFoundError:No module named 'tensorflow.contrib'

这是由于tensorflow v2不再支持contrib模块,可以搜索v1版本下contrib中函数在v2中对应的名称,然后进行替换。
但是,由于我的源代码含有较多contrib模块的函数,修改较为麻烦,因此放弃这种方法。


解决方案:

由于原有方法兼容性较差,因此采用安装tensorflow v1的方法。

注意:无法在python 3.9 中使用pip install无法直接安装tensorflow 1.x,可以将conda环境下的python版本降到3.6(vscode和pycharm不再支持调试python3.5,其他版本没有尝试)。查看conda环境的Python版本

采用博客中的方法安装,如下所示,将python=3.5改成python=3.6即可。

物联沃分享整理
物联沃-IOTWORD物联网 » 【已解决】使用TensorFlow报错:“ModuleNotFoundError: No module named ‘tensorflow.contrib’”

发表评论