CUDNN error: CUDNN_STATUS_EXECUTION_FAILED解决方法

本人在ubuntu18.04系统复现执行cyclegan网络过程,使用显卡GeForce RTX 3090,该显卡仅仅支持使用cuda11以上的版本。

由于之前配置官网指导环境pytorch1.4.0,误选用cuda10.1及cuda10.2,导致每次运行test.py过程极慢,半天过后报错CUDNN error: CUDNN_STATUS_EXECUTION_FAILED,

后来,更换cuda11,对应的pytorch版本也就高于1.4.0了,程序才运行飞起,究其原因,在于显卡本身不支持cuda11以下版本,无语凝噎,也就说这里所谓的版本不匹配,是显卡硬件型号与cuda本身版本不匹配导致的报错,Mark一下,希望能对大家有所用处。

ps:更换新版本的pytorch的cuda及cudnn安装教程见我上一篇博客分享

你可能感兴趣的:(深度学习,pytorch,pycharm)