新闻资讯
利用Conda安装深度学习框架Pytorch
1.前言
Pytorch目前是炙手可热的深度学习框架。和TensorFlow比较起来学习曲线更加平滑,不用写大量的样板代码就可以对网络进行训练和使用。在最新版本的Pytorch中开始支持Java的。但是安装Pytorch并不是很容易的事。今天就来说一下如何利用Conda安装Pytorch。
2.康达
这里简单提一下Conda,Conda是一个开源的,跨平台的互连管理系统和环境管理系统,用于安装多个Python版本的替代及其依赖关系,并在它们之间轻松切换。你可以把康达看作Python的领域的Maven的,当然有些功能可能比Maven的更加强大。通常我会安装蟒蛇来集成康达和Python的环境,anacoda额外还有numpy的,熊猫等有用的科学计算包。如果你有洁癖也可以使用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下和窗口平台类似。
回复列表