[tensorflow][keras]Failed to get convolution algorithm.解决方法Failed to get convolution algori

tenorflow-gpu 1.14或者1.13等版本经常有如下错误，这个不是你cuda安装有问题，而是显存不足造成，提示错误如下

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

 

对于tensorflow而言，代码前面加入下面代码可以解决问题

from tensorflow.compat.v1 import ConfigProto
 from tensorflow.compat.v1 import InteractiveSessionconfig = ConfigProto()
 config.gpu_options.allow_growth = True
 session = InteractiveSession(config=config)

如果使用keras则可以用下面代码解决问题

import tensorflow as tf
 import numpy as np
 import keras
 config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
  
 config.gpu_options.per_process_gpu_memory_fraction = 0.3
 tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))

如果是tensorflow2.x需要用下面代码解决问题

 

import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
 for gpu in gpus:
     tf.config.experimental.set_memory_growth(gpu, True)