首页 > 编程知识 正文

为什么pytorch慢,pytorch卡住了

时间:2023-05-03 23:54:25 阅读:204920 作者:396

pytorch下载(如何解决pytorch下载慢的问题)

本文主要致力于提供pytorch的解决方法。

我是在anaconda的环境下安装的。如果没有安装anaconda的可以先去官网安装个anaconda:anaconda官网。

1.正常安装pytorch
先去官网pytorch官网找到要下载的对应版本。

这里建议安装stable的版本。还要注意需不需要GPU加速,需要的话要看自己电脑支持的是哪个版本,可以去NVIDIA设置中查看,我这里选择的是10.2。
用anaconda安装的话建议使用conda进行安装(pip我试了下。。好像不行。。)
选择好了以后就复制下面的command。
打开Anaconda Prompt,将command粘贴,回车。
这个方法非常的慢,我试了好几次都因为太慢了中断了好几次。
然后我查到可以使用清华镜像源进行安装,速度会比刚才这个方法快上很多!

2.使用清华镜像源安装pytorch
第一步:打开Anaconda Prompt。
第二步:更改channels设置,依次执行以下命令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/mainconda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/freeconda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/rconda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloudconda config --set show_channel_urls yes

第三步:我们还是到pytorch的官网,复制对应版本的command,然后删除后面的-c pytorch,因为-c pytorch默认从pytorch官网下,是非常慢的。删除-c pytorch之后,执行该命令,可以发现这个方法安装要快上很多。

下载完之后我们可以验证一下是否下载成功:
可以在当前环境下输入python命令,然后输入import torch看是否报错;我们用conda list命令也可以查看是否安装成功,可以发现里面已经有pytorch及torchvision等等。

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。