使用conda卸载pytorch_利用 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

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

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 获取更多干货

你可能感兴趣的:(使用conda卸载pytorch_利用 Conda 安装深度学习框架 Pytorch)