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

2,814 阅读2分钟

1. 前言

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

2. Conda

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

5. 总结

今天介绍了如果利用 Conda 安装 Pytorch ,解决 Pytorch 安装失败的一些问题。如果你在安装中有什么不明白可以通过私信我。

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

个人博客:https://felord.cn