我在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