如何在Windows和Linux上安装PyTorch

在本指南中,我将介绍如何在Windows和Linux上安装PyTorch,并使用Anaconda来管理软件包的安装。

安装将使用内置在Anaconda中的conda命令行工具进行。

PyTorch是一个用Python编写的库,基于Torch框架。它由Facebook开发,可与Google的Tensorflow等相媲美。它在计算机视觉和自然语言处理领域非常有用,并已被特斯拉等公司用于开发自动驾驶软件。

PyTorch是免费且开源的,根据修改后的BSD许可证,并受Linux Foundation管理。

先决条件

要按照本教程操作,您需要在使用的计算机上安装Anaconda。

如果您尚未安装它,可以使用此指南how to install Anaconda全面介绍安装过程。在按照该指南操作后,您可以继续安装PyTorch。

在Linux上安装PyTorch

作为良好的实践,首先更新Linux发行版中的软件包。在我的情况下,我使用Ubuntu和apt来管理我的软件包,因此我将使用以下命令进行更新:

$ sudo apt update && apt upgrade 

当更新软件包完成后,转到官方网站PyTorch website installation page。向下滚动页面,直到找到下面的安装向导:

使用此向导,您可以单击不同的选项,提供系统信息和首选项,从而获得一个可在终端中使用的命令来安装PyTorch。

在提供系统信息后,看起来是这样的:

我选择了稳定的Linux版本,并且将使用Conda来管理我的软件包。我还选择了在Python而不是C++/Java中使用PyTorch。我将在CPU而不是GPU上运行PyTorch。

在表格底部是我可以使用的命令来安装PyTorch,但在运行此命令之前,我想创建一个名为pytorch的Anaconda虚拟环境。

虚拟环境允许您创建项目并使其依赖项与其他项目的依赖项隔离,从而防止依赖冲突。Anaconda的好处之一是它可以帮助您轻松创建和管理虚拟环境。

要创建一个Python版本为3.7的虚拟环境,我将输入以下命令:

conda create -n pytorch python=3.7

创建环境后,我将使用以下命令激活它:

conda activate pytorch

环境激活后,我将在PyTorch网站上生成的命令来安装PyTorch。

conda install pytorch torchvision torchaudio cpuonly -c pytorch

按照提示安装PyTorch。完成后,我将重新启动终端会话以生效。

现在,要验证PyTorch是否已正确安装,我们将尝试在Python交互式shell中导入它。确保您处于PyTorch虚拟环境中,使用以下命令:

conda activate pytorch

一旦您进入了PyTorch虚拟环境,通过输入以下命令打开Python交互式shell:

python

启动shell会话后,输入以下代码并按ENTER键

import torch

如果Python运行没有错误,则安装成功。但是,如果出现”Module Not Found”错误,则表示安装过程出错。您可以尝试重新安装。

在Windows上安装PyTorch

首先,在您的Windows机器上搜索Anaconda Prompt程序并打开它。这是我们将要运行命令的地方。

程序打开后,我们将使用以下命令为我们的PyTorch安装创建一个虚拟环境。

conda create -n pytorch python=3.7

创建虚拟环境后,我们可以通过运行以下命令来激活它:

conda activate pytorch

虚拟环境激活后,我们可以继续安装PyTorch。我们首先前往 PyTorch website installations page。之后,我们可以滚动到页面的这个安装向导所在的部分:

在这里,我们选择我们的系统信息,然后向导将给出一个安装PyTorch的命令。我将选择Windows的稳定版本,由Conda管理,在Python编程语言中使用,以及在CPU上运行。因此,我的命令如下所示。

接下来,复制这个命令,将其粘贴到Anaconda提示符中,然后按ENTER键。

安装完成后,我们可以通过打开Python交互式shell并尝试导入PyTorch来验证安装是否成功。

因此,在Anaconda提示符内启动一个交互式Python会话。

python

会话启动后,使用以下代码行导入PyTorch:

import torch

如果此操作完成而没有错误,则表示安装成功。

最后的话

在本指南中,我们使用conda在Windows和Linux上安装了PyTorch。也可以像普通的PIP软件包一样通过PIP进行安装。在这两种情况下,我选择了CPU安装。但是,您仍然可以使用CUDA,这是由Nvidia开发的系统工具包,通过在GPU上并行化操作来加快训练速度。

类似文章