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

1 问题

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

2 方法

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

04aa98c3d70042085e976658d2f73621.png找到对应的Anaconda版本下载即可。

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

94df9356202a843e0e228419c71db57c.png

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

f933c9350e84e75b39d2d8b7a2577524.png

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

a4a6ed02d569bfc876d56ea16ea5a4b6.png

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

在AnacondaPrompt界面输入:

c76e4aa0b4fa4218cda7fb9be4929cc2.png

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

4c2ca0c717cb4af547f8f0d072f58ec8.png

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

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

7e02e55d62f8fe5dc8ea4bf11b6b54a0.png

3 结语

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

凌寒、任宗伟、钟源

2022年9月9日

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

算法与编程之美

欢迎关注『算法与编程之美』

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值