解决RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

在我用服务器跑神经网络时,出现了这个问题,反复在网上查找解决方案,感觉网络上的方案无法解决问题,于是开始自己尝试解决。

实际原因

出现这个问题的实际原因应该是,显卡与pytorch版本不一致,导致无法正常调用cuda进行数据处理
在我这个服务器上,RTX2080并没有更新驱动以适配pytorch.1.7.1,然而我的pytorch是1.8.0

解决方案

更改pytorch为1.2.0,问题解决

你可能感兴趣的:(pytorch,cuda)