问题描述:

运行基于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 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐