UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [Op:Conv2D]
解决办法:代码前添加下面两行
import os
os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘/gpu:0’
具体为什么完全搞不懂
来源:简书
作者:白小白db
链接:https://www.jianshu.com/p/e13aa15f35da