conda安装GPU版pytorch,却是cpu版本的原因
原因用官网的命令conda安装GPU版本pytorch,发现是安装的cpu版本。查看了下库的信息,发现有一个pytorch-mutex库,这个就是原因所在,将其卸载后,再重新安装GPU版本就解决了。卸载conda uninstall pytorch-mutex安装重新安装后,问题解决。conda install pytorch torchvision torchaudio cudatoolkit=
·
已为社区贡献1条内容
所有评论(0)