在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
软件包。