网络编程
位置:首页>> 网络编程>> Python编程>> 教你两步解决conda安装pytorch时下载速度慢or超时的问题

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

作者:Polaris_T  发布时间:2022-04-12 17:19:59 

标签:conda,安装,pytorch,超时

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):

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

3.激活虚拟环境

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

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

4.安装pytorch

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

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

下面是重点:
在安装命令后加上–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

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

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

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

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

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

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

来源:https://blog.csdn.net/qq_45717425/article/details/122432586

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com