conda install pytorch 安装缓慢的解决办法

2019/08/11更新

根据评论,清华镜像已恢复,请移步至清华 conda 镜像使用方法。

在国内镜像停止支持 conda 后,安装速度巨慢,尤其是安装 pytorch时,简直蛋痛。

在折腾了一晚上后,摸索出一可行的解决方法:手动下载所需文件放至 conda 缓存目录,解压,然后再运行 conda install 命令直接安装。

下文以 Linux 安装 pytorch 为例,作为 step-by-step 备忘。

第一步:找到 conda 的缓存目录

输入:

conda info

《conda install pytorch 安装缓慢的解决办法》
《conda install pytorch 安装缓慢的解决办法》

得到缓存路径,例中为:/home/zjh/anaconda3/pkgs

第二步:手动下载所需文件

运行:

conda install torchvision cudatoolkit=10.0 -c pytorch

得到所需下载的 pytorch 版本: pytorch-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0.tar.bz2

《conda install pytorch 安装缓慢的解决办法》
《conda install pytorch 安装缓慢的解决办法》

进入 https://anaconda.org/pytorch/pytorch/files 寻找对应文件并下载

《conda install pytorch 安装缓慢的解决办法》
《conda install pytorch 安装缓慢的解决办法》

此时可以使用各种下载工具,使用科学上网服务器走代理下载速度更快。

《conda install pytorch 安装缓慢的解决办法》
《conda install pytorch 安装缓慢的解决办法》

第三步:将 pytorch-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0.tar.bz2 解压至 conda 缓存目录

将文件放至 conda 缓存路径,本例中由第一步知路径为:/home/zjh/anaconda3/pkgs

运行以下命令,创建解压目录并解压:

mkdir pytorch-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0
tar -jxvf pytorch-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0.tar.bz2 -C pytorch
-1.1.0-py3.6_cuda10.0.130_cudnn7.5.1_0

(注:若压缩包名为 some-package.tar.bz2,则解压目录为 some-package)

《conda install pytorch 安装缓慢的解决办法》
《conda install pytorch 安装缓慢的解决办法》

解压完成后,在 /anaconda3/pkgs 下的 urls.txt 文件内添加此 package 的下载路径。

(此步骤非常重要,若不在此文件添加pacakge下载路径,则 conda 会再次下载文件)(这个结论是我实践得出来的,没有 document 参考,欢迎指正或补充)

《conda install pytorch 安装缓慢的解决办法》
《conda install pytorch 安装缓慢的解决办法》

至此,已完成手工下载所需文件,并放置 conda 缓存目录,直接运行 conda install 进行安装即可。

《conda install pytorch 安装缓慢的解决办法》
《conda install pytorch 安装缓慢的解决办法》

由上图发现,无需下载 pytorch 文件,直接进行安装。

《conda install pytorch 安装缓慢的解决办法》
《conda install pytorch 安装缓慢的解决办法》

测试 torch 包,可正常使用。

此解决方法手工下载所需,省去 conda 漫长的 torch 包下载时间。虽然略有麻烦,但很省时间。如有更方便的方法,还请您分享给大家。

    原文作者:快乐肥宅赵同学
    原文地址: https://zhuanlan.zhihu.com/p/73741240
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞