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)

你可能感兴趣的:(keras: 使用CuDNNSLTM TypeError: Fail to find the dnn implementation.)