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，
·
本人在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安装教程见我上一篇博客分享
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
6
0- 0
已为社区贡献1条内容
所有评论(0)