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

2020-05-28 15:57:31

1. 前言

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

2. Conda

这里简单提一下 Conda ，Conda 是一个开源的、跨平台的软件包管理系统和环境管理系统，用于安装多个 Python 版本的软件包及其依赖关系，并在它们之间轻松切换。你可以把 Conda 看作 Python 领域的 Maven ，当然有些功能可能比 Maven 更加强大。通常我会安装 anaconda 来集成 Conda 和 Python 环境，anacoda 额外还有 numpy 、pandas 等有用的科学计算包。如果你有洁癖也可以使用 MiniConda ，它只包含 Conda 和 Python 。如果你还不会 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.-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.-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.

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

以上图中的版本为例，下载成功后执行以下命令分别安装 pytorch 和 torchvision 包：

   #  env_name  为 Conda 环境名
   conda install --offline  env_name /path/to/pytorch-1.4.-py3.7_0.tar.bz2
   conda install --offline  env_name /path/to/torchvision-0.5.-py37_cpu.tar.bz2

这样 Pytorch 的安装就完成了，你可以愉快地进行炼丹了。linux 和 windows 平台类似。