C/C++教程

Windows下利用Anaconda下载Pytorch1.9.1并使用GPU11.1【清华源】

本文主要是介绍Windows下利用Anaconda下载Pytorch1.9.1并使用GPU11.1【清华源】,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

1.安装Anaconda(略)

        此步骤一定要确保安装成功并能打开Anaconda Navigator

2.更改Anaconda官方源

        打开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/

3.配置Python环境

       1. 创建环境

conda create -n pytorch python=3.9

        其中pytorch和3.9分别为环境名称和版本号,可自行更改

        2.激活环境

conda activate pytorch

        看到(pytorch)出现即为成功

4.安装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已安装完毕

这篇关于Windows下利用Anaconda下载Pytorch1.9.1并使用GPU11.1【清华源】的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!