cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

在服务器上跑CNN网络代码时出错。
先查看python环境中torch和cuda版本。

import torch
print(torch.__version__)
print(torch.version.cuda)

结果显示
torch 1.8.0 cuda 10.2
看网上说可能是torch和cuda版本不匹配。于是我删除conda虚拟环境中的pythoch环境。

conda remove pytorch

接着重新安装torch

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

安装后发现问题并没解决。
后来到pytorch官方网站重新安装最新版。https://pytorch.org/

问题解决了，深度学习程序成功在服务器上运行成功，并跑出来了预期结果。