解决cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

前配置：python 3.7.11   cuda9.0   pytorch1.0.0及对应cudnn   rtx2060

跑一个程序总报这个错误。

解决方法：

这是我摸索到的三种解决方法

1、换成CPU模式就能跑，但速度慢。

2、在代码前设置 torch.backends.cudnn.enabled = False也能跑，但速度慢。也可能报cublas runtime error: the GPU program failed to execute显存不够的错误。但我觉得不至于，所以有了第三种方法。

3、好像rtx20系列用CUDA 10.0比较好，于是我将CUDA换成10.0。

https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

pytorch版本去官网找对应的。我用的是pytorch1.2.0 Previous PyTorch Versions | PyTorch​​​​​

结果可以完美跑了。如果还不行，那可能你显存确实不够吧。