解决PyTorch下载安装慢的问题

PyTorch是Facebook在2017年1月发布的一款深度学习框架。近些年来深度学习的发展速度迅猛,从之前谷歌的AlphaGo到如今OpenAI的ChatGPT,一波一波的浪潮席卷着国内的深度学习研究者们,开发者们也开发了很多框架。随着不断的淘汰优化,市面上主流使用的是TensFlow和PyTorch这两个框架。在之前的一篇文章里,我已经详细说过了如何在没有NVIDIA显卡的情况下用Anconda3安装PyTorch。但是由于PyTorch官网在国外的服务器上,我们如果要进行下载,数据的传输会经过漫长的海底光缆,因此下载的速度会很慢,时常会出现安装太慢,进度条不变的情况。接下来我会教大家下载PyTorch时,如何提速。(注意:我已经默认读者装好了Anconda3,系统是Windows)

一、换源

国内有一些大学,比如清华大学有自己的镜像源库,既然从国外网站下的慢,不如直接国内的镜像源下载。首先,要找到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/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

二、查看.condarc文件

进入C盘,在C盘的Users里找到.condarc文件,以记事本打开来,查看是否有default一行。如果有,请删除,不然还是会默认下载。最终的结果应该是和我一样的。

三、进入PyTorch官网查看下载命令

根据以下的红色选项,找到下载命令。

在刚才打开的Anaconda Prompt窗口里先激活PyTorch环境

conda activate pytorch

在名称为pytorch环境下输入在官网选择的命令,注意去掉pytorch官网给的 -c pytorch

conda install pytorch torchvision torchaudio cpuonly 

就是说官网的命令是conda install pytorch torchvision torchaudio cpuonly -c pytorch

你自己要用国内镜像源下载得是conda install pytorch torchvision torchaudio cpuonly ,这样才会在安装的时候才会默认从镜像源下载相应的包。

四、查看下载情况

下载的时候,初始配置环境很慢,不要觉得没配置好,请一定要保持耐心,我就这样半途而废过。

安装PyTorch的时候,如果命令行出现清华镜像源的地址,说明配置有效,安心下载吧。

最后下好后,测试安装是否成功的过程可以参考我的上一篇文章。Windows下用Anaconda3安装PyTorch环境攻略(针对AMD显卡)

展开阅读全文

页面更新:2024-04-17

标签:清华大学   清华   显卡   框架   深度   命令   情况   环境   文件   国内

1 2 3 4 5

上滑加载更多 ↓
推荐阅读:
友情链接:
更多:

本站资料均由网友自行发布提供,仅用于学习交流。如有版权问题,请与我联系,QQ:4156828  

© CopyRight 2020-2024 All Rights Reserved. Powered By 71396.com 闽ICP备11008920号-4
闽公网安备35020302034903号

Top