如何通过conda安装适配tensorflow-gpu 1.6.0的cuDNN解决版本不匹配报错
适配tensorflow-gpu 1.6.0的CuDNN conda安装方案
报错本质是运行环境加载的CuDNN 7.6.5版本,和tensorflow-gpu 1.6.0编译时绑定的CuDNN 7.1.x版本不匹配,严格对齐版本即可解决。
操作步骤
- 先清理现有不兼容的CuDNN版本
- 执行命令查看当前环境的CuDNN安装情况:
- Windows:
conda list | findstr cudnn - Linux/macOS:
conda list | grep cudnn
- Windows:
- 卸载现有高版本CuDNN:
conda remove cudnn
- 执行命令查看当前环境的CuDNN安装情况:
- 检索适配的历史版本
- 老版本CuDNN需从anaconda官方频道检索,执行命令:
conda search -c anaconda cudnn=7.1.* - 从返回结果中选择依赖对应CUDA 9.0的版本(通常7.1.4是tensorflow-gpu 1.6.0的最优匹配版本)
- 老版本CuDNN需从anaconda官方频道检索,执行命令:
- 安装指定版本的依赖
- 执行命令同时安装对齐的CuDNN和配套CUDA,避免conda自动升级到不兼容的高版本:
conda install -c anaconda cudnn=7.1.4 cuda=9.0
- 执行命令同时安装对齐的CuDNN和配套CUDA,避免conda自动升级到不兼容的高版本:
- 安装后验证
- 执行
conda list确认输出的cudnn版本为7.1.x、cuda版本为9.0,无其他高版本CuDNN残留 - 重新运行代码即可解决卷积算子加载失败的报错
- 执行
注意事项
- tensorflow-gpu 1.6.0 仅支持CuDNN 7.1.x版本,不兼容7.2及以上的所有CuDNN版本,不要直接安装最新版依赖
- 如果安装时出现依赖冲突提示,可添加
--override-channels参数强制从anaconda频道拉取历史包,避免conda自动拉取其他频道的高版本依赖导致版本错位
内容的提问来源于stack exchange,提问作者kikicoder
相关产品推荐
相关产品推荐

