Python升级tensorflow2.x版本相关问题：No module named ‘tensorflow.contrib‘ 问题解决

Posted 2023-03-06 挣扎的蓝藻

tensorflow2.x 版本舍去了 contrib 相关功能，相关功能的使用可以通过安装 tf_slim 来引用。

Traceback (most recent call last):
  File "test.py", line 2, in <module>
    from tools.utils import *
  File "C:\\Users\\Administrator\\Desktop\\AnimeGANv2-master\\tools\\utils.py", line 2, in <module>
    from tensorflow.contrib import slim
ModuleNotFoundError: No module named 'tensorflow.contrib'

解决方法1：
可以将 tensorflow 版本更换为 1.x 系列的老版本。

解决方法2：
安装 tf_slim，安装方法 pip install tf_slim

然后按目录找到对应报错的文件。

将代码里引入时的 tensorflow.contrib 修改为 tf_slim 即可。
下面实例代码是将 from tensorflow.contrib import slim
修改为 import tf_slim as slim 的演示。

还有个报错是这个：

Traceback (most recent call last):
  File "test.py", line 8, in <module>
    from net import generator
  File "C:\\Users\\Administrator\\Desktop\\AnimeGANv2-master\\net\\generator.py", line 1, in <module>
    import tensorflow.contrib as tf_contrib
ModuleNotFoundError: No module named 'tensorflow.contrib'

同样的方法修改一下即可。

然后就解决了。
喜欢的点个赞❤吧。