【已解决】使用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即可。
更多推荐
已为社区贡献1条内容
所有评论(0)