Failed to get convolution algorithm. This is probably because cuDNN failed to initialize解决办法

试过改变cuddn版本,但是没啥效果。感觉应该是显存的问题,如下修改:

tensorflow:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as session:

keras:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))

我的环境是cuddn7.6 + cuda10.0 + python3.6 2080ti这样改就可以运行了

你可能感兴趣的:(解决办法,cuddn,tensorflow,keras)