【已解决】使用tensorflow报错:ModuleNotFoundError:No module named ‘tensorflow.contrib‘
tensorflow
一个面向所有人的开源机器学习框架
项目地址:https://gitcode.com/gh_mirrors/te/tensorflow
免费下载资源
·
问题描述:
运行基于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即可。
GitHub 加速计划 / te / tensorflow
184.54 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:1 个月前 )
a49e66f2
PiperOrigin-RevId: 663726708
2 个月前
91dac11a
This test overrides disabled_backends, dropping the default
value in the process.
PiperOrigin-RevId: 663711155
2 个月前
更多推荐
已为社区贡献1条内容
所有评论(0)