神經網絡訓練時出現 無法獲取卷積算法的問題 tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize

 

 

 

使用訓練環境：

Tensorflow 2.2.0，CUDA 10.1，cuDNN 7.6.5。

當實例化VGG16卷積基時，出現了無法獲取卷積算法的問題，提示可能因為cuDNN無法初始化。

打開代碼文件在前面添加如下行，問題得以解決。

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)