win10环境下,深度学习pytorch出现:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

配置:微星2060显卡 + cuda10.0 + torch1.1.0 + torchvision0.0.3 + python3.6.5

首先:

      根据网上教程安装好了anaconda,python版本为3.6.5;接着根据显卡驱动,这里我选择安装CUDA10和对应的CUDNN,这里注意一点,在CUDA安装过程中,记得去掉VS的勾选,不然会报错;一切进行正常,在pycharm中利用GPU跑深度模型时出现这个错误:“  RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED ”;

接着:

       网上开始refer各种方案,但是win10环境下出现这种问题的解决方案真的很少;重复卸载安装CUDA、anaconda多次,结果依然让人无奈;于是我考虑更换CUDA版本(CUDA9.0、CUDA9.1),仍无济于事。有人说是python版本的原因,我尝试过python3.6.6依然报错。(是不是很蛋疼!!!)

       最后重点来了,继续在CUDA10和对应CUDNN下,我在anaconda3.6.5下创建了虚拟环境:python3.7+ torch1.2 + torchvision0.4;这里更新了torch 和 torchvision的版本。最后完美解决上述问题,真是心快死了。

 Note:

       通过官方torch的方法conda、pip安装torch太慢了,我选择了下载 .whl文件的形式进行安装。

  致谢:

同样踩坑博客:https://blog.csdn.net/qq_39938666/article/details/86611474?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522159963040519724836718894%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=159963040519724836718894&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~first_rank_v2~rank_v28-1-86611474.pc_first_rank_v2_rank_v28&utm_term=Runtime+Error+cuDnn+error&spm=1018.2118.3001.4187

 

 

你可能感兴趣的:(深度学习,深度学习,anaconda,cuda,python)