RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS（cuda,pytorch版本不匹配的问题）

问题描述

在服务器上运行个人的代码，服务器是大家公用的，所以cuda不方便更改或者添加新的版本，服务器上的cuda版本为10.1.105。我的代码使用的torch=0.4.1版本，对应的cuda我感觉是9.0左右，不是十分确定。但10.1的是肯定不能。

解决方法

当时看了相关的博客感觉是版本不匹配的问题，所以重新开了虚拟环境，采用pytorch官网的相关命令：conda install pytorch=0.4.1 torchvision cudatoolkit=10.1 -c pytorch进行强制更新安装。不过最后还是运行错误。可能的原因感觉是0.4版本可能只能支持到cuda=9的程度，再高的版本就没办法。
最后将torch版本改为1.0就好，只是相关的在0.4下的方法或者说是写法就会变得非法，需要进行部分的更改。