在Ubuntu上,如何安装PyTorch

分享于 

7分钟阅读

ubuntu

 

通过Anaconda安装

步骤1:安装Anaconda

Anaconda是很受欢迎 包管理系统,适用于使用Python的数据科学家,这样做的原因是,因为它预装了完整的数据科学包,它是PyTorch推荐的软件包管理界面。

首先,最好确保所有软件包都是最新的:

root@ubuntu1604:~# apt-get update -y

在此之后,我们需要下载并运行 bash Anaconda的安装脚本。

注: 不建议运行直接从internet下载的bash脚本,在继续之前,请确保你信任提供安装脚本的源,下面curl命令中的链接是撰写本文时Linux命令行安装程序的最新版本,
root@ubuntu1604:~# curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

安装脚本下载完成后,运行它,并按照提示操作,我们要用默认值

root@ubuntu1604:~# bash Miniconda3-latest-Linux-x86_64.sh

要开始使用Anaconda,我们需要刷新终端:

root@ubuntu1604:~# source ~/.bashrc

默认情况下,每次你输入新的shell会话时,它将自动激活,要关闭此行为,可以运行以下命令:

(base) root@ubuntu1604:~# conda config --set auto_activate_base false

出于本教程的目的,我们希望允许anaconda环境的基础继续处于活动状态。

第2步:安装PyTorch

现在我们已经安装并激活了Anaconda,现在该安装PyTorch了。

(base) root@ubuntu1604:~# conda install pytorch torchvision cpuonly -c pytorch

你会在安装过程中看到提示,请输入" y"以完成安装。

Proceed ([y]/n)? y

验证PyTorch是否已安装

最后,是时候验证是否已安装PyTorch并 python repl (a


(base) root@ubuntu1604:~# Python
Python 3.7.4 (default, Aug 13 2019, 20:35:49)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>>

进入repl之后,我们可以粘贴以下代码片段,


import torch
x = torch.rand(3, 7)
print(x)

输出应该类似于以下内容:


>>> import torch
>>>
>>> x = torch.rand(3, 7)
>>> print(x)
tensor([[0.2833, 0.7547, 0.1444, 0.6286, 0.4838, 0.1576, 0.3350],
    [0.8544, 0.5095, 0.0808, 0.4367, 0.4174, 0.4669, 0.6155],
    [0.4436, 0.7904, 0.8911, 0.2073, 0.5987, 0.3607, 0.4366]])
>>>

要退出Python Shell,请按住ctrl键并按D键(Ctrl +D)。

现在我们回到命令提示符处,如果使用,


(base) root@ubuntu1604:~# conda deactivate
root@ubuntu1604:~#

通过PIP安装

步骤1 :安装Python3-venv

如果你不需要Anaconda附带的所有其他软件包,你可以在虚拟Python环境中使用Python软件包管理器Pip安装PyTorch ,为了确保PyTorch的安装及它依赖项对系统的Python安装没有不利影响,建议将它安装在虚拟Python环境中。

首先,我们需要安装python3-venv软件包,以便创建虚拟Python环境。

root@ubuntu1604:~# apt-get install -y python3-venv

步骤2:准备环境

首先,创建一个目录来容纳 项目 使用cd命令将其更改为:


root@ubuntu1604:~# mkdir pytorch_awesome
root@ubuntu1604:~# cd pytorch_awesome
root@ubuntu1604:~/pytorch_awesome#

现在是时候创建一个虚拟Python环境,我们在其中安装PyTorch,

root@ubuntu1604:~/pytorch_awesome# python3 -m venv pytorch-awesome

root@ubuntu1604:~/pytorch_awesome# source pytorch-awesome/bin/activate
(pytorch-awesome) root@ubuntu1604:~/pytorch_awesome#

现在我们可以看到在普通的shell提示符之前出现了"(pytorch-awesome )",因为新创建的虚拟Python环境已激活。

步骤3:安装PyTorch

在新的Python虚拟环境处于活动状态时,我们可以安装PyTorch 。

(pytorch-awesome) root@ubuntu1604:~/pytorch_awesome# pip install torch==1.3.0+cpu torchvision==0.4.1+cpu -f https://download.pytorch.org/whl/torch_stable.html

我们可以像安装Anaconda一样测试PyTorch是否已正确安装。

注: 你在Python Shell的初始输出中看到的Python版本可能与我的略有不同

(pytorch-awesome) root@ubuntu1604:~/pytorch_awesome# Python
Python 3.5.2 (default, Oct 8 2019, 13:06:37)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>>

现在复制并粘贴这段代码到 shell并点击enter :


import torch
x = torch.rand(3, 7)
print(x)

输出将类似于以下内容:


>>> import torch
>>>
>>> x = torch.rand(3, 7)
>>> print(x)
tensor([[0.0416, 0.9980, 0.2793, 0.8503, 0.0285, 0.8286, 0.8091],
    [0.8038, 0.7944, 0.0110, 0.1239, 0.0611, 0.9727, 0.6899],
    [0.3036, 0.0378, 0.1660, 0.7076, 0.5073, 0.0686, 0.9490]])
>>>

要退出Python Shell,请按住ctrl键并按D键(Ctrk +D)。

回到命令提示符后,我们可以停用Python虚拟环境。


(pytorch-awesome) root@ubuntu1604:~/pytorch_awesome# deactivate
root@ubuntu1604:~/pytorch_awesome#

Ubuntu