此步骤一定要确保安装成功并能打开Anaconda Navigator
打开Anaconda Prompt
添加清华镜像站的channel,否则下载速度极其缓慢
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
conda create -n pytorch python=3.9
其中pytorch和3.9分别为环境名称和版本号,可自行更改
conda activate pytorch
看到(pytorch)出现即为成功
1.进入Pytorch官网,根据CUDA版本号选取所需命令 关于CUDA版本号,要看个人显卡的配置
打开NVIDIA控制面板->帮助->系统信息->组件
笔者为11.1,故选11.1版本(若为11.1则清华源一定要添加conda-forge地址)
接下来复制官网上提供的命令
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
注意此时要把后面的两个“-c”全部删去,它们的意思是,安装下载的channel强制为pytorch官网的channel,所以需要删除才能走清华镜像的channel,所以我们需要输入的命令是
conda install pytorch torchvision torchaudio cudatoolkit=11.1
稍作等待后,会显示需要下载的各种文件,按照提示输入y,回车,开始下载
漫长的等待后,下载完成,显示done
接下来按照图示操作检验pytorch是否安装正确
其中最后一条命令为检验显卡是否支持
至此,pytorch已安装完毕