Windows环境下tensorflow报错“Could not load library cudnn_cnn_infer64_8.dll. Error code 126”的解决方案

Table of Contents

问题描述

Windows环境下tensorflow的报错信息为:

Could not load library cudnn_cnn_infer64_8.dll. Error code 126
Please make sure cudnn_cnn_infer64_8.dll is in your library path!

我的cuda版本为11.5,cudnn版本是与之对应的11.5。

解决方案

我重装了cuda 11.5与对应的cudnn,反复确认二者版本一致,但不能解决问题。
查找cuda路径,cudnn_cnn_infer64_8.dll文件是存在的,但tensorflow无法导入。
该文件负责CNN底层运算,因此该问题不解决,模型无法运行。
有两种解决方案:

  • 在保持cuda版本为11.5的情况下,安装cuDNN v8.2.2 (July 6th, 2021), for CUDA 11.4。原问题是由于cuDNN与cuda不匹配导致的。为什么官方的版本还是不匹配?或许这就叫bug吧!
  • 使用WinDLL(path)手工导入该文件路径,但并不推荐这一方式,因为每当需要运行CNN时都需要手动导入一遍。

需要说明的一点是,如果你的cuda版本不是11.5但依旧出现了这一问题,请检查如下两点:

  • 你的显卡是否支持当前的cuda版本
  • cuda版本是否与cudnn版本匹配

版权声明:本文为博主乔卿原创文章,版权归属原作者,如果侵权,请联系我们删除!

原文链接:https://blog.csdn.net/qq_41112170/article/details/121878288

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
xiaoxingxing的头像xiaoxingxing管理团队
上一篇 2022年2月17日 下午3:11
下一篇 2022年2月17日 下午4:26

相关推荐

此站出售,如需请站内私信或者邮箱!