原因:cuda版本和pytorch版本不匹配
遇到这个问题,一开始很困惑,明明安装的torch版本是1.12啊,是匹配cuda11.3的啊,但是同一个torch版本也是有多个cuda编译版本的,所以一定要指定安装cuda11.3编译的那个,可以按照torch官网的指令安装:
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113