Windows系统下有英伟达显卡安装PyTorch

Posted 2022-09-17 算法与编程之美

1 问题

学习深度学习技术需要使用到PyTorch，在安装PyTorch 过程中，有很多的细节需要知道，在下载PyTorch的时候，我们需要根据英伟达显卡的版本进行相应的下载，如何查看英伟达显卡版本？下载Anaconda的时候很慢，如何让Anaconda下载很快？在使用Anaconda下载PyTorch非常慢，如何使用Anaconda下载PyTorch加快？

2 方法

我们在官网下载Anaconda很慢，我们就可以使用清华大学开源软件镜像网站进行下载，网址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

找到对应的Anaconda版本下载即可。

查看英伟达显卡支持cuda的版本：打卡cmd,输入nvidia-smi,可以看到英伟达显卡支持的cuda版本。

根据cuda版本去Pytorch官网下载对应的版本

安装好Anaconda后，就打开AnacondaPrompt,在Pytorch找到相应的版本

复制Run this Command这一列的代码，在AnacondaPrompt界面输入这行代码，就开始下载pytorch,因为AnacondaPrompt用的是国外的服务器，所以下载很慢，我们就可以添加清华的镜像源，这样下载就会很快了。

在AnacondaPrompt界面输入：

然后在c盘弹出的.condarc配置文件，用于修改下载源，在该配置文件中写入：

就成功换了清华的镜像源，下载pytorch就很快了。

下载完pytorch，我们可以验证一下是否成功，如果是true就成功了

3 结语

针对pytorch的安装的一系列问题，如果我们电脑里有英伟达的显卡，并且支持cuda,那么在安装pytorch的过程中，我们就要注意版本的问题的，我们需要下载自己电脑上cuda对应的pytorch版本。

凌寒、任宗伟、钟源

2022年9月9日