欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

解决 Anaconda 安装 pytorch 下载速度慢的问题

程序员文章站 2022-07-14 09:38:08
...

问题描述:

在官网按照教程敲写语句:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

解决 Anaconda 安装 pytorch 下载速度慢的问题
但是这里有个很严重的问题,就是这里安装渠道是走pytorch官网的,所以会是十分的慢。因此,针对这个下载pytorch速度很慢的问题给出以下的解决方法。

解决方法

第一种方法,清华源下载相关的包

第一步,去清华源那里下载对应的pytorch包,比如我是python3.7的,需要pytorch-gpu版本的,找到对应的bz2文件下载,如下图。(注意这个是windows版本的,linux或者mac请返回到上一层寻找对应系统)
解决 Anaconda 安装 pytorch 下载速度慢的问题
第二步,下载好就在命令行进入你下载的路径目录里面,并输入:

conda install pytorch-XXXX(你的下载文件名字).bz2

第三步,命令行里面输入:

conda install torchvision

第四步,安装结束,去试试吧

第二种方法,搭建清华源

第一步,在命令行输入换一个下载源:

# ubuntu
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

第二步,直接输入安装命令:

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

第三步,完事,这个方法很棒,我在ubuntu上可行。

参考链接:
github大佬们的回答