Windows,Linux和MacOS上的PyTorch安装

时间:2020-02-23 14:43:42  来源:igfitidea点击:

PyTorch的安装非常简单,可以在所有主要操作系统上完成。
但是,如果您想在不进行实际安装的情况下弄脏手,则Google Colab提供了一个很好的起点。

Colab随附了预安装的PyTorch和Tensorflow模块,并与GPU和TPU支持一起使用。

为了在您自己的计算机上安装,PyTorch随附CUDA和不提供CUDA版本,具体取决于您可用的硬件。

这将是一个单步安装– PyTorch从本地启动。

前提条件:Anaconda发行版(指向官方的链接)–您需要在系统上安装Anaconda才能遵循本教程。
下载包适用于所有主要操作系统,安装过程非常简单。

因此,在继续学习本教程之前,请确保已在操作系统上建立并运行了Anaconda发行版。

注意:如果您不想使用Anaconda,则可以始终使用PIP安装PyTorch。
由于PIP与Python安装程序捆绑在一起,因此您已经将其安装在系统中。

在Windows上安装Pytorch

PyTorch为Windows系统提供以下命令。
PyTorch可在Windows 7或者更高版本上使用,并使用Python 3或者更高版本。
使用Anaconda进行安装非常简单,只需几分钟即可完成。

PyTorch安装1

下一步是将以下命令粘贴到Anaconda提示符中并运行它。

conda install pytorch torchvision cpuonly -c pytorch

该提示将列出将与PyTorch一起安装的所有依赖项。
如果可以继续,请在命令行中输入yes。

Anaconda现在继续安装。
您可以稍后通过Python解释器或者Jupyter Notebook检查安装。

现在,您已经在Windows系统上成功安装了PyTorch。

在Linux上安装PyTorch

如果从Linux机器上打开相同的安装页面,您会注意到生成的命令将是一个不同的命令。

下一步是将命令复制并粘贴到终端中并运行它。

conda install pytorch torchvision cpuonly -c pytorch

终端会要求您许可安装/更新软件包。
您需要按是。

现在,安装将继续将Torch和Torchvision软件包安装到您的环境中。

在Mac OS上安装PyTorch

我们将使用PIP在Mac OS中安装PyTorch。
我们需要做的就是在PyTorch主页上选择适当的选项以获取安装命令。

Pytorch Quick Start Locally Mac Os

因此,我们必须运行以下命令在Mac OS上安装PyTorch和Torchvision库。

$pip3.7 install torch torchvision

这是执行上述命令时来自终端的输出。

输出的最后一行清楚地表明,torch和torchvision软件包都已成功安装。

让我们启动Python 3.7解释器并打印割炬版本以确认安装成功。

$python3.7
Python 3.7.3 (v3.7.3:ef4ec6ed12, Mar 25 2019, 16:52:21) 
[Clang 6.0 (clang-600.0.57)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
1.4.0
>>>