python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > 解决安装pytorch下载慢

教你两步解决conda安装pytorch时下载速度慢or超时的问题

作者:Polaris_T

这篇文章主要介绍了教你两步解决conda安装pytorch时下载速度慢or超时的问题,使用清华镜像源可以大大减少安装的时间,需要的朋友可以参考下

1.为conda配置清华源

打开cmd输入以下命令:

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 --set show_channel_urls yes

2.创建新的虚拟环境(若已创建,请忽略)

如果你是第一次安装Miniconda,那么只存在base环境。由于还要安装Pytorch,所以要再创建一个新的虚拟环境。我们可以利用如下命令创建环境:

conda create -n xxxx(环境名) python=x.x(python版本) 

例如:

conda create -n pytorch python=3.6

创建完成后,在命令行输入conda env list就能看到目前已经创建的环境(我的环境名为gluon,所以下图中可以看到除了base外还有个gluon):

在这里插入图片描述

3.激活虚拟环境

在命令行输入conda activate 环境名即可激活环境,如下图所示:

在这里插入图片描述

4.安装pytorch

在pytorch官网中选择你要安装的pytorch版本,下方就会给出相应的安装命令。比如这里我用的是win10+cuda11.3并且选择用pip安装:

在这里插入图片描述

下面是重点:
在安装命令后加上–default-timeout=1000,修改下载超时判定的条件为1s(这么干不仅降低了超时的概率,似乎还提高了下载速度(?玄学))
比如我用的是:

pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio===0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html --default-timeout=1000

输入命令后敲个回车就等它自己慢慢下载了…

在这里插入图片描述

最后在Python中import一下然后打印一下torch版本号,验证是否安装成功:

在这里插入图片描述

或者在虚拟环境中使用pip list显示所有已安装的包,可以找到torch及torchvision:

在这里插入图片描述

到此这篇关于教你两步解决conda安装pytorch时下载速度慢or超时的问题的文章就介绍到这了,更多相关解决安装pytorch下载慢内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文