RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

Huntersts

于 2020-08-13 13:29:45 发布

阅读量3k

点赞数 3

本文链接：https://blog.youkuaiyun.com/qq_35260390/article/details/107977451

版权

在升级到Ubuntu 18.04和CUDA 10.0后，遇到cuDNN错误CUDNN_STATUS_EXECUTION_FAILED。尝试过更换cudatoolkit、调整torch和Python版本无效。最终解决方案是创建新的conda环境，使用pytorch=1.4，torchvision=0.2.1，python=3.6.10，cudatoolkit=10.0.130，并修改cudnn设置。此外，发现cuDNN版本和CUDA版本匹配，关闭CUDNN.BENCHMARK也可避免错误。

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

在GTX1650+cuda10.0+Ubuntu18.04环境中运行pytorch出现CUDNN_STATUS_EXECUTION_FAILED的错误的一个解决办法

网上关于这个问题的解决方法有很多,有的是更换cudatoolkit,还有的是更换torch和Python的版本的,但是这些在我这里并不适用.
最初我的程序里关于cudnn是这样设置的,当时是使用Ubuntu16.04+cuda9+pytorch1.1

cudnn.benchmark = False
torch.backends.cudnn.deterministic = False
torch.backends.cudnn.enabled = True

后来更换了Ubuntu18.04+cuda10.0之后就一直提示各种错误,最开始是提示libcudart.so.9.0找不到路径,然后我更换cudatoolkit之后就开始报这个cuDNN error: CUDNN_STATUS_EXECUTION_FAILED错误.

最终我对cuDNN error: CUDNN_ST

最低0.47元/天解锁文章