我之前使用conda install命令安装gpu版本的pytorch的,如下:
然后就出现了上面的问题,看了下别人的博客,好像如果conda配置了清华镜像源,然后conda install安装的就是cpu版本的torch,不是gpu版本的,我也不知道是不是这个问题
解决办法:
先把安装的torch卸载了,然后用pip安装gpu版本的torch(我没用镜像源,我开了外网下好像),
即执行命令下面即可
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121