新闻资讯

新闻资讯 通知公告

利用Conda安装深度学习框架Pytorch

编辑:016     时间:2020-02-15

1.前言

Pytorch目前是炙手可热的深度学习框架。和TensorFlow比较起来学习曲线更加平滑,不用写大量的样板代码就可以对网络进行训练和使用。在最新版本的Pytorch中开始支持Java的。但是安装Pytorch并不是很容易的事。今天就来说一下如何利用Conda安装Pytorch

2.康达

这里简单提一下CondaConda是一个开源的,跨平台的互连管理系统和环境管理系统,用于安装多个Python版本的替代及其依赖关系,并在它们之间轻松切换。你可以把康达看作Python的领域的Maven的,当然有些功能可能比Maven的更加强大。通常我会安装蟒蛇来集成康达Python的环境,anacoda额外还有numpy的,熊猫等有用的科学计算包。如果你有洁癖也可以使用MiniConda,它只包含CondaPython。如果你还不会Conda建议你花十分钟快速入门一下。

3.在线安装Pytorch

Mac为例,Conda安装最新版Pytorch的命令如下:

conda install pytorch torchvision -c pytorch 复制代码

根据该命令的提示很容易就能安装好Pytorch如果安装失败,很可能是因为网络的延迟过高而超时所以我们可以通过配置anacoda国内镜像源来加速安装命令执行以下命令

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 --set show_channel_urls yes 复制代码

然后重新安装Pytorch即可。

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 为Pytorch的普遍源

4.离线安装Pytorch

有时甚至使用过多的源也会失败,所以我们可以通过离线安装来解决这个问题。当Pytorch安装失败后命令行终端如果出现以下提示:

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/osx-64/pytorch-1.4.0-py3.7_0.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/osx-64/torchvision-0.5.0-py37_cpu.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way. 复制代码

这种情况一般是pytorchtorchvision 包下载失败,我们可以选择手动下载它们然后离线安装。我们从上面的提示中提取包下载url链接进行下载,你可以将下载url链接中的 https://conda.anaconda.org/pytorch/替换为https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/以提高下载速度。

以上插图的版本为例,下载成功后执行以下命令分别安装 pytorchtorchvision 包:

#  env_name  为 Conda 环境名 conda install --offline  env_name /path/to/pytorch-1.4.0-py3.7_0.tar.bz2
   conda install --offline  env_name /path/to/torchvision-0.5.0-py37_cpu.tar.bz2 复制代码

这样Pytorch的安装就完成了,你可以愉快地进行炼丹了。linux下窗口平台类似。

郑重声明:本文版权归原作者所有,转载文章仅为传播更多信息之目的,如作者信息标记有误,请第一时间联系我们修改或删除,多谢。

回复列表

相关推荐