最近在实验室服务器上配置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
总算是成功了。