环境:
Anaconda3-5.2.0
python3.6.5
初始tensorflow版本是1.8.0,cuda9.0,升级tensorflow至1.10.0时,出现torch1.1.0的运行错误:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED,刚开始以为是升级tensorflow时导致cuda某些内容被更新,所以尝试了好几种搭配未能解决问题,反而导致CUDA报错,有关版本的库(**.dll)找不到,后来查找原因发现是版本不匹配导致,最后证明有效搭配如下:
tensorflow1.14.0+torch1.2.0+CUDA10.0
本人亲测有效,这边贴出百度云的链接,需要的朋友可以直接下载:
- Anaconda3-5.2.0
链接:https://pan.baidu.com/s/15zsZQZ_-4Ein4lvNAOFCkg 提取码:vgk1
- tensorflow-gpu 1.14.0
链接:https://pan.baidu.com/s/1HsRu3Z4QsUEqg6AEkBebmQ 提取码:bmcb
- pytorch-gpu 1.2.0
链接:https://pan.baidu.com/s/1i_MZoP-aPPsgNQMhlF_5Jg 提取码:gjs2
- CUDA10.0+CUDnn
链接:https://pan.baidu.com/s/1kG_5i7LgToxVeL0G4_LujA 提取码:jwfw
- CUDA10.2+CUDnn
链接:https://pan.baidu.com/s/1U-R1EmJlEfICKJL6ASlejw 提取码:utd7
转载请注明出处,希望对需要的人有所帮助