博客主要讲述了Keras使用CuDNNLSTM时出现TypeError: Fail to find the dnn implementation错误的解决办法，即调用前添加代码设置gpu_options.allow_growth = True，并给出了具体的代码示例。
keras: 使用CuDNNSLTM TypeError: Fail to find the dnn implementation.
再调用前加一行代码，设置gpu_options.allow_growth = True
import tensorflow as tf
from tensorflow import keras
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config = config)
keras.backend.set_session(sess)
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
