win10+keras 跑ResNet 完美解決Failed to get convolution algorithm. This is probably because cuDNN failed to initialize報錯

#GPU按需分配,解決
import tensorflow as tf
import keras
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))

參考https://blog.csdn.net/qq_41868689/article/details/98503069

原因是顯存不足，因此設置成按需分配。