模型訓練錯誤：Failed to get convolution algorithm，This is probably because cuDNN failed to initialize.

一、問題：

         最近在利用keras訓練模型的時候出現以下錯誤，開始一看以為是cuDNN的版本不對，更換后還是沒有效果，后面在網上查找后發現是顯存的問題，更改為動態分配內存就可以了。

 

  二、解決方法：

         在訓練的腳本開頭添加以下代碼進行動態分配內存，然后重新運行就可以。

 

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))

 

三、運行環境

•  keras: 2.2.4
• tensorflow-gpu: 1.14.0
• CUDA: 10.0.130
• cudnn: 7.6.3