python,win10,tensorflow2.20。oaded runtime CuDNN library: 7.4.1 but source was compiled with: 7.6.5.

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

结果:成功医治

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值