ModuleNotFoundError: No module named ‘tensorflow.contrib‘ 解决方案

问题描述

今天在执行以下代码时：

from tensorflow.contrib.layers.python import layers as tf_layers

出现了以下错误：

ModuleNotFoundError: No module named 'tensorflow.contrib' 

看网上说应当是使用tensorflow2.x运行tensorflow1.x的代码导致的。由于cuda版本和python版本的关系，降低tensorflow的版本是不可行的，因此这里尝试其他的方案。 

解决方案

首先安装tensorflow-slim（官网地址）：

pip install --upgrade tf_slim

然后将上述报错语句改为：

from tf_slim import layers as tf_layers

 完美解决问题。

参考链接

Why can I not import Tensorflow.contrib I get an error of No module named 'tensorflow.python.saved - Stack Overflow