在网上查了很多资料，首先要确定是不是版本问题，
像我在虚拟环境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
