python3.6+pytorch1.9.0 + cuda11.1+cuDNN v8.1.0 + tensorflow2.4.0+tensorflow_gpu-2.4.0

呜呜呜呜呜呜花了好几天这个破环境终于解决了。

报错:Loaded runtime CuDNN library: 8.0.5 but source was compiled with: 8.1.0.  CuDNN library needs to have matching major version and equal or higher minor version. If using a binary install, upgrade your CuDNN library.  If building from sources, make sure the library loaded at runtime is compatible with 
the version specified during compile configuration.

解决方法:将pytorch和tensorflow共同用到的CuDNN版本匹配一致。将pytorch升级到8.1.0后系统还是报错,一直检测不到新升级的CuDNN,环境配置也改了。最后将tensorflow所用到的CuDNN降到8.0.5解决上述问题。

查看对应版本:2022最新:TensorFlow各个GPU版本CUDA和cuDNN对应版本整理(最简洁)_hi_1900的博客-CSDN博客_tensorflowgpu版本对应关系

我的环境为:python3.6+pytorch1.9.0 + cuda11.1+cuDNN v8.0.5 + tensorflow2.4.0+tensorflow_gpu-2.4.0

安装命令:

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html 

pip install tensorflow==2.4.0

pip install tensorflow-gpu==2.4.0

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值