解决RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

机器学习(自然语言处理)

于 2021-08-26 11:02:56 发布

阅读量748

点赞数

文章标签： pytorch 深度学习 cuda

本文链接：https://blog.csdn.net/m0_58287747/article/details/119926903

版权

博主在配置Ubuntu18.04系统、GTX2080Ti和RTX3080显卡上，遇到CUDA9.0、CUDNN7.1.2和PyTorch0.4.0运行错误。尝试了多种版本组合，最终发现RTX1080Ti与相同环境匹配能成功运行。问题可能在于新显卡与旧版驱动的不兼容，提示在版本匹配时还需考虑显卡驱动的适配性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

配置环境耽误了一天的时间，有必要记录一下以后少跳坑。

刚开始尝试的环境是：系统：Ubuntu 18.04、显卡：GTX 2080Ti、cuda9.0、cudnn7.1.2、pytorch0.4.0,其中cuda、cudnn和pytorch版本是从官网对应下来的，版本匹配肯定没问题。

我是用的 conda install pytorch=0.4.0 cuda=9.0 cudnn=7.1.2 来安装的。其中python版本是3.6.9（比较稳定，不建议用太新的），然后程序一开始运行就报错了，错误出现在/lib/python3.6/site-packages/torch/nn/模块，就是cudnn加速循环神经网络这块。我尝试着更换cuda/cudnn/pytorch/python各种版本，最终都无效。

然后把显卡类型换到了RTX3080，也是以上各种操作，最终都无果。我寻思着，cuda/cudnn/pytorch版本是不是太老了，在比较新的显卡上有问题，顺着这个想法走了走，把显卡换成了RTX1080Ti,环境匹配和上面的2080Ti的一摸一样，结果可以运行了，解决！

出现错误不只是版本匹配问题，当版本匹配时，还报错，可能是显卡太新/驱动太高。