Tensorflow常见问题——Tensorflow 2.X中import tensorflow.contrib.rnn as rnn错误不降级的解决方法

@TOCTensorflow 2.X中import tensorflow.contrib.rnn as rnn错误不降级的解决方法

前言

笔者在gitee上面下载了基于深度学习的滚动轴承故障诊断方法（[https://gitee.com/dongfangxiaotian/Fault_Diagnosis.git]，发现其运行环境为tensorflow==1.8，但是我下载的是tensorflow2.X版本的。运行其中的cnn.py时发现import tensorflow.contrib.rnn as rnn语句报错，错误类型为ModuleNotFoundError: No module named ‘tensorflow.contrib‘；

    import tensorflow.contrib.rnn as rnn
ModuleNotFoundError: No module named 'tensorflow.contrib'

错误原因

查询其他文章发现这是由于tensorflow2.X里面不包含tensorflow.contrib包。

解决方法

1. 将自己的tensorflow将为1.x版本，但是过程繁琐，容易出问题；
2. 不降级解决方法：
   （1）将import tensorflow as tf 改为

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

（2）将 import tensorflow.contrib.rnn as rnn改为from tensorflow.python.ops import rnn
运行结果：

参考

错误ModuleNotFoundError: No module named ‘tensorflow.contrib‘的解决方法
ModuleNotFoundError: No module named ‘tensorflow.contrib’ 解决方法
tensorflow2报错AttributeError: module ‘tensorflow‘ has no attribute ‘variable_scope‘