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

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.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.

复制代码

这种状况通常是 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.0-py3.7_0.tar.bz2
   conda install --offline  env_name /path/to/torchvision-0.5.0-py37_cpu.tar.bz2
复制代码

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

5. 总结

今天介绍了若是利用 Conda 安装 Pytorch ，解决 Pytorch 安装失败的一些问题。若是你在安装中有什么不明白能够经过私信我。

关注公众号：Felordcn获取更多资讯

我的博客：https://felord.cn