【Tensorflow+keras】解决 Fail to find the dnn implementation.
1 引言(1)环境Tensorflow 2.0Python 3.6(2)问题from tensorflow.keras import *from tensorflow.keras.layers import Bidirectional,LSTM在使用Bidirectional(LSTM )时,报错[Derived]Fail to find the dnn implementation.2 解决因为
·
1 引言
(1)环境
Tensorflow 2.0
Python 3.6
(2)问题
from tensorflow.keras import *
from tensorflow.keras.layers import Bidirectional,LSTM
在使用Bidirectional(LSTM )时,报错 [Derived] Fail to find the dnn implementation.
2 解决
因为在Tensorflow2.0中使用keras的API时,该版本的keras已经不支持直接使用Bidirectional(LSTM ),三种解决办法。
(1)第一种方法
继续使用在Tensorflow2.0的keras API,用兼容的替代的写法
from tensorflow.keras import *
from tensorflow.keras.layers import Bidirectional
Bidirectional(tf.keras.layers.RNN(tf.keras.layers.LSTMCell(n_BiLSTM_1),return_sequences=True))
(2)第二种方法
使用支持该方法的原装Keras:2.1.6
# 安装
pip install keras ==2.1.6
# 使用
from keras.layers import Bidirectional,LSTM
(3)第三种方法
继续使用在Tensorflow2.0的keras API,但加入表示兼容的代码
from tensorflow.keras.layers import Bidirectional,LSTM
from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)
开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!
更多推荐
已为社区贡献14条内容
所有评论(0)