我用的是windows下的miniconda，cuda版本是10.0，安装的pytorch版本是1.7，怀疑是高版本的pytorch不兼容cuda10.0。成功解决办法：
安装cuda10.2和对应cudnn
删除miniconda3pkgs文件夹下cuda、pytorch、torchvision等相关的文件夹
重新创建虚拟环境
conda create -n pytorch_1.7 python=3.6
在虚拟环境中安装pytorch
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
