问题描述
今天在执行以下代码时:
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
完美解决问题。
参考链接