Linux环境下安装Pytorch默认安装位CPU版本解决方法

最近在实验室服务器上配置Pytorch时被折磨。使用Pytorch官网的指令

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

但下载时发现安装的是cpu版本。

但当我通过conda uninstall pytorch卸载后，无论安装哪个版本的pytorch都会默认下载cpu版本。

最终解决方法

输入

conda uninstall pytorch-mutex

会卸载掉cpuonly和pytorch-mutex两个模块，之后再选择1.8版本的pytorch成功安装gpu版本。指令如下：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.1

总算是成功了。