我之前使用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
Logo

一站式 AI 云服务平台

更多推荐