问题：Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

问题：Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed ab

keras报以下错误：

tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[{{node vgg16/block1_conv1/convolution}}]] [[{{node loss/mul}}]]

解决：

在train.py等训练文件下增加以下代码：

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

tensorflow报以下错误：

UnknownError (see above for traceback): Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[node FirstStageFeatureExtractor/resnet_v1_101/resnet_v1_101/conv1/Conv2D (defined at /home/zgq/detector_faster-rcnn/nets/resnet_utils.py:122) ]] [[node BalancedPositiveNegativeSampler_3/Where_1 (defined at /home/zgq/detector_faster-rcnn/object_detection/core/minibatch_sampler.py:80) ]]

解决：

在train.py等训练文件下增加以下代码：

import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)

报如下错误均可尝试用以上方法解决

Failed to get convolution algorithm. This is probably because cuDNN failed to initialize