欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

解决cannot import name ‘CuDNNLSTM‘和cannot import name ‘CuDNNGRU‘

程序员文章站 2022-04-25 20:22:59
...

原代码:

from keras.layers import CuDNNGRU, CuDNNLSTM, SpatialDropout1D

报错:

ImportError:cannot import name ‘CuDNNGRU‘

分开引入:

from keras.layers import CuDNNGRU
from keras.layers import CuDNNLSTM
from keras.layers import SpatialDropout1D

报错:

ImportError:cannot import name ‘CuDNNGRU‘
ImportError:cannot import name ‘CuDNNLSTM‘

SpatialDropout1D正常引入

最后改为:

from tensorflow.compat.v1.keras.layers import CuDNNGRU as GRU
from tensorflow.compat.v1.keras.layers import CuDNNLSTM as LSTM
from keras.layers import SpatialDropout1D

运行正常。

参考文章:解决cannot import name ‘CuDNNLSTM‘问题

                  ImportError: cannot import name 'XXX' 问题解决方案

相关标签: 毕业设计 Python