遇到ModuleNotFoundError:当我使用Tensorflow GPU处理时,没有名为'tensorflow.contrib'的模块

时间:2019-11-24 22:48:35

标签: tensorflow keras tensorflow2.0 cudnn

ModuleNotFoundError: No module named 'tensorflow.contrib'的行from tensorflow.contrib.cudnn_rnn.python.ops import cudnn_rnn_ops中执行keras\layers\cudnn_recurrent.py命令时,我遇到425。此问题特定于Tensorflow 使用GPU处理时。如果我不使用GPU处理,那就没有问题。

这是我的软件堆栈:

 - OS: Windows 10
 - Python: 3.7.4
 - CUDA: 10.1
 - cuDNN: 7.5.0.56
 - tensorflow: 2.0.0
 - tensorflow-gpu: 2.0.0
 - keras: 2.3.1

编辑:为了更清楚一点,我为了使用GPU处理(因为我的GPU支持CUDA)而不使用CPU处理而在代码中所做的唯一更改是使用keras.layers.CuDNNLSTM类而不是keras.layers.LSTM 。当然,我已经通过tensorflow-gpu安装了pipenv软件包。

1 个答案:

答案 0 :(得分:1)

TensorFlow 2.0已停止支持Contrib。您可以进一步了解here

可能是您的代码是为TF 1. *编写的,或者未正确移植。