cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

在网上查了很多资料，首先要确定是不是版本问题，
像我在虚拟环境pip安装用pip list查看的cuda和torch分别是10.2.89和1.5.0
但是在cuda list里面是10.2.89和1.0.0
很怪，无法理解
但是在python中打印出cuda和torch又分别是9.0和1.0.0
5555好窒息，搞了一整天很暴躁

conda list

在python中打印
无法理解


conda list和pip list的版本不同又和python打印出来的不同，而且我配置的torchvision是0.4的，肯定不能支持cuda9.0
具体可参考 https://www.zhihu.com/question/395145313/answer/1260811851

好吧，强行重装cuda和torch

conda install pytorch=1.5.0 cudatoolkit=10.2.89 -c pytorch

再次在python里面打印出来，

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

成功！再次运行，报错解除！

最终环境是
python3.7.11
cuda 10.2.89
cudnn 7.6.5
torch 1.5.0
torchvision 0.4