ensorflow.python.framework.errors_impl.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.
[[node model_1/conv2d/Conv2D (defined at d:/my_life/yolo_learn/xixixi/yolov4-tiny-tf2-master/train.py:439) ]] [Op:__inference_train_function_10810]
Function call stack:
train_function
以上为报错,具体如下图
错误原因分析:如报错所示,cudnn版本出现问题。
问题描述:笔者在前几天安装了tensorflow1.13.2版本和cuda10.0,以及10.0对应的cudnn。在此之前笔者环境中装有tensorflow==2.2.0和cuda10.1,以及cudnn7.6.5在c盘。几天前两种版本互不干扰,运行融洽(有兴趣可以找笔者两天前刚发的文章)。但是今天刚打开tensorflw2.2.0训练文件,出现了以上的错误,显示cudnn版本出现问题。
解决方法:依旧采取conda虚拟环境配置cuda和cudnn的方法
1.activate tensorflow2
2.conda install cudnn=7.6.5
结果:成功医治