导致的原因一般都是显卡算力和cuda或者torch版本不匹配
由于我是自己去下载的torch和torchvision可能没下好对应的关系,于是我卸载了相对应的torch和torchvision,运行下面的命令 安装了对应的版本
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
导致的原因一般都是显卡算力和cuda或者torch版本不匹配
由于我是自己去下载的torch和torchvision可能没下好对应的关系,于是我卸载了相对应的torch和torchvision,运行下面的命令 安装了对应的版本
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch