最近在實驗室服務器上配置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
總算是成功了。