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

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

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

當然，出現這種問題也存在cudnn版本問題，但是發現這種情況並不多，大多是上面這種。