RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

运行Pytorch出现如题报错，网上找的方法基本都认为时cudnn，cuda等某一个版本问题，但是没有给出很好的解决方案，清一色重装，我找到另一个方法，比较准确的定位了问题，首先开命令行运行以下代码

import torch
import torch.nn as nn

rnn = nn.RNN(10, 20, 2)
rnn.cuda()

此时出现了更详细的报错

RuntimeError: cuDNN version mismatch: PyTorch was compiled against 7102 but linked against 7605

这里面，我现在的cudnn的版本是7.6.5,对应了7605，而安装的pytorch0.4.1时基于cudnn7.1.2编译的，因此重装7.1.2版本的cudnn即可

conda install cudnn=7.1.2