快捷方式

    本地启动

    选择您的偏好设置并运行安装命令。稳定版代表 PyTorch 当前测试和支持的最新版本。这应该适合许多用户。如果您想要最新的、尚未完全测试和支持的 nightly 构建版本,则可以使用预览版。请确保您已满足以下先决条件(例如,numpy),具体取决于您的包管理器。Anaconda 是我们推荐的包管理器,因为它会安装所有依赖项。您也可以安装以前的 PyTorch 版本。请注意,LibTorch 仅适用于 C++。

    注意:最新版本的 PyTorch 需要 Python 3.9 或更高版本。

    PyTorch 版本
    您的操作系统
    软件包
    语言
    计算平台
    运行此命令
    PyTorch 版本
    稳定版 (1.13.0)
    预览版 (Nightly)
    您的操作系统
    Linux
    Mac
    Windows
    软件包
    Conda
    Pip
    LibTorch
    源代码
    语言
    Python
    C++ / Java
    计算平台
    CUDA 11.8
    CUDA 12.1
    CUDA 12.4
    ROCm 5.2
    CPU
    运行此命令
    conda install pytorch torchvision -c pytorch


    在 macOS 上安装

    PyTorch 可以在 macOS 上安装和使用。根据您的系统和 GPU 功能,您在 Mac 上使用 PyTorch 的体验在处理时间方面可能会有所不同。

    先决条件

    macOS 版本

    PyTorch 支持 macOS 10.15(Catalina)或更高版本。

    Python

    建议您使用 Python 3.9 - 3.12。您可以通过 Anaconda 包管理器(请参见下文)、HomebrewPython 网站安装 Python。

    包管理器

    要安装 PyTorch 二进制文件,您需要使用以下两个支持的包管理器之一:pipAnaconda

    Anaconda

    要安装 Anaconda,您可以下载图形安装程序或使用命令行安装程序。如果您使用命令行安装程序,您可以右键单击安装程序链接,选择复制链接地址,或在配备 Apple 芯片的 Mac 计算机上使用以下命令

    # The version of Anaconda may be different depending on when you are installing`
    curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh
    sh Miniconda3-latest-MacOSX-arm64.sh
    # and follow the prompts. The defaults are generally good.`
    

    pip

    Python 3

    如果您通过 Homebrew 或 Python 网站安装了 Python,则pip 会随其一起安装。如果您安装了 Python 3.x,则将使用命令pip3

    提示:如果您只想使用命令pip,而不是pip3,您可以将pip 符号链接到pip3 二进制文件。

    安装

    Anaconda

    要通过 Anaconda 安装 PyTorch,请使用以下 conda 命令

    conda install pytorch torchvision -c pytorch
    

    pip

    要通过 pip 安装 PyTorch,请使用以下两个命令之一,具体取决于您的 Python 版本

    # Python 3.x
    pip3 install torch torchvision
    

    验证

    为了确保 PyTorch 已正确安装,我们可以通过运行 PyTorch 示例代码来验证安装。在这里,我们将构造一个随机初始化的张量。

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

    输出应类似于

    tensor([[0.3380, 0.3845, 0.3217],
            [0.8337, 0.9050, 0.2650],
            [0.2979, 0.7141, 0.9069],
            [0.1449, 0.1132, 0.1375],
            [0.4675, 0.3947, 0.1426]])
    

    从源代码构建

    对于大多数 PyTorch 用户而言,通过包管理器从预构建的二进制文件安装将提供最佳体验。但是,有时您可能希望安装最新的 PyTorch 代码,无论是用于测试还是在 PyTorch 核心上进行实际开发。要安装最新的 PyTorch 代码,您需要从源代码构建 PyTorch

    先决条件

    1. [可选] 安装Anaconda
    2. 请按照此处描述的步骤操作:https://github.com/pytorch/pytorch#from-source

    您可以按照上文所述验证安装。

    在 Linux 上安装

    PyTorch 可以在各种 Linux 发行版上安装和使用。根据您的系统和计算需求,您在 Linux 上使用 PyTorch 的体验在处理时间方面可能会有所不同。建议(但不是必须)您的 Linux 系统配备 NVIDIA 或 AMD GPU,以便充分利用 PyTorch 的CUDA支持ROCm支持。

    先决条件

    支持的 Linux 发行版

    PyTorch 支持使用glibc >= v2.17 的 Linux 发行版,包括以下发行版

    此处的安装说明通常适用于所有支持的 Linux 发行版。例如,您的发行版可能支持yum 而不是apt。显示的具体示例是在 Ubuntu 18.04 计算机上运行的。

    Python

    Python 3.9-3.12 通常默认安装在任何支持的 Linux 发行版上,这符合我们的建议。

    提示:默认情况下,您必须使用命令python3 来运行 Python。如果您只想使用命令python,而不是python3,您可以将python 符号链接到python3 二进制文件。

    但是,如果您想安装其他版本,有多种方法

    如果您决定使用 APT,则可以运行以下命令进行安装

    sudo apt install python
    

    如果您使用Anaconda安装 PyTorch,它将安装一个沙盒版本的 Python,该版本将用于运行 PyTorch 应用程序。

    包管理器

    要安装 PyTorch 二进制文件,您需要使用以下两个支持的包管理器之一:Anacondapip。Anaconda 是推荐的包管理器,因为它将在一个沙盒安装中为您提供所有 PyTorch 依赖项,包括 Python。

    Anaconda

    要安装 Anaconda,您将使用命令行安装程序。右键单击 64 位安装程序链接,选择复制链接位置,然后使用以下命令

    # The version of Anaconda may be different depending on when you are installing`
    curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
    sh Miniconda3-latest-Linux-x86_64.sh
    # and follow the prompts. The defaults are generally good.`
    

    您可能需要打开一个新的终端或重新加载您的~/.bashrc 才能访问conda 命令。

    pip

    Python 3

    虽然 Python 3.x 默认安装在 Linux 上,但pip 并非默认安装。

    sudo apt install python3-pip
    

    提示:如果您只想使用命令pip,而不是pip3,您可以将pip 符号链接到pip3 二进制文件。

    安装

    Anaconda

    无 CUDA/ROCm

    要通过 Anaconda 安装 PyTorch,并且您的系统没有CUDA 功能ROCm 功能,或者不需要 CUDA/ROCm(即 GPU 支持),在上面的选择器中,选择操作系统:Linux、软件包:Conda、语言:Python 和计算平台:CPU。然后,运行显示给您的命令。

    使用 CUDA

    要通过 Anaconda 安装 PyTorch,并且您的系统具有CUDA 功能,在上面的选择器中,选择操作系统:Linux、软件包:Conda 和适合您机器的 CUDA 版本。通常,最新版本的 CUDA 更好。然后,运行显示给您的命令。

    使用 ROCm

    目前,Anaconda 中的 PyTorch 不支持 ROCm。请改用 pip。

    pip

    无 CUDA

    要通过 pip 安装 PyTorch,并且没有 支持 CUDA支持 ROCm 的系统,或者不需要 CUDA/ROCm(即 GPU 支持),在上面的选择器中,选择 OS:Linux,Package:Pip,Language:Python 以及 Compute Platform:CPU。然后,运行显示给您的命令。

    使用 CUDA

    要通过 pip 安装 PyTorch,并且拥有 支持 CUDA 的系统,在上面的选择器中,选择 OS:Linux,Package:Pip,Language:Python 以及适合您机器的 CUDA 版本。通常,最新的 CUDA 版本更好。然后,运行显示给您的命令。

    使用 ROCm

    要通过 pip 安装 PyTorch,并且拥有 支持 ROCm 的系统,在上面的选择器中,选择 OS:Linux,Package:Pip,Language:Python 以及支持的 ROCm 版本。然后,运行显示给您的命令。

    验证

    为了确保 PyTorch 已正确安装,我们可以通过运行 PyTorch 示例代码来验证安装。在这里,我们将构造一个随机初始化的张量。

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

    输出应类似于

    tensor([[0.3380, 0.3845, 0.3217],
            [0.8337, 0.9050, 0.2650],
            [0.2979, 0.7141, 0.9069],
            [0.1449, 0.1132, 0.1375],
            [0.4675, 0.3947, 0.1426]])
    

    此外,要检查您的 GPU 驱动程序和 CUDA/ROCm 是否已启用并可供 PyTorch 访问,请运行以下命令以返回 GPU 驱动程序是否已启用(PyTorch 的 ROCm 版本在 Python API 级别上使用相同的语义 链接,因此以下命令也适用于 ROCm)

    import torch
    torch.cuda.is_available()
    

    从源代码构建

    对于大多数 PyTorch 用户而言,通过包管理器从预构建的二进制文件安装将提供最佳体验。但是,有时您可能希望安装最新的 PyTorch 代码,无论是用于测试还是在 PyTorch 核心上进行实际开发。要安装最新的 PyTorch 代码,您需要从源代码构建 PyTorch

    先决条件

    1. 安装 AnacondaPip
    2. 如果您需要构建支持 GPU 的 PyTorch:a. 对于 NVIDIA GPU,如果您的机器具有 支持 CUDA 的 GPU,请安装 CUDA。b. 对于 AMD GPU,如果您的机器具有 支持 ROCm 的 GPU,请安装 ROCm
    3. 请按照此处描述的步骤操作:https://github.com/pytorch/pytorch#from-source

    您可以按照 上面 所述验证安装。

    在 Windows 上安装

    PyTorch 可以在各种 Windows 发行版上安装和使用。根据您的系统和计算需求,您在 Windows 上使用 PyTorch 的体验在处理时间方面可能会有所不同。建议(但不是必须)您的 Windows 系统配备 NVIDIA GPU,以便充分利用 PyTorch 的 CUDA 支持

    先决条件

    支持的 Windows 发行版

    PyTorch 支持以下 Windows 发行版

    此处的安装说明通常适用于所有支持的 Windows 发行版。显示的具体示例将在 Windows 10 Enterprise 机器上运行

    Python

    目前,Windows 上的 PyTorch 仅支持 Python 3.9-3.12;不支持 Python 2.x。

    由于 Python 不是 Windows 上的默认安装项,因此有多种方法可以安装它。

    如果您使用 Anaconda 安装 PyTorch,它将安装一个沙盒版本的 Python,该版本将用于运行 PyTorch 应用程序。

    如果您决定使用 Chocolatey,并且尚未安装 Chocolatey,请确保以管理员身份运行命令提示符。

    对于基于 Chocolatey 的安装,请在管理员命令提示符下运行以下命令

    choco install python
    

    包管理器

    要安装 PyTorch 二进制文件,您需要使用至少两种受支持的包管理器之一:Anacondapip。Anaconda 是推荐的包管理器,因为它将在一个沙盒安装中为您提供所有 PyTorch 依赖项,包括 Python 和 pip.

    Anaconda

    要安装 Anaconda,您将使用 PyTorch 3.x 的 64 位图形安装程序。点击安装程序链接并选择 运行。Anaconda 将下载,并会显示安装程序提示。默认选项通常是合理的。

    pip

    如果您通过任何推荐的方式 上面 安装了 Python,则 pip 已经为您安装好了。

    安装

    Anaconda

    要使用 Anaconda 安装 PyTorch,您需要通过 开始 | Anaconda3 | Anaconda Prompt 打开 Anaconda 提示符。

    无 CUDA

    要通过 Anaconda 安装 PyTorch,并且没有 支持 CUDA 的系统,或者不需要 CUDA,在上面的选择器中,选择 OS:Windows,Package:Conda 以及 CUDA:None。然后,运行显示给您的命令。

    使用 CUDA

    要通过 Anaconda 安装 PyTorch,并且您拥有 支持 CUDA 的系统,在上面的选择器中,选择 OS:Windows,Package:Conda 以及适合您机器的 CUDA 版本。通常,最新的 CUDA 版本更好。然后,运行显示给您的命令。

    pip

    无 CUDA

    要通过 pip 安装 PyTorch,并且没有 支持 CUDA 的系统,或者不需要 CUDA,在上面的选择器中,选择 OS:Windows,Package:Pip 以及 CUDA:None。然后,运行显示给您的命令。

    使用 CUDA

    要通过 pip 安装 PyTorch,并且您拥有 支持 CUDA 的系统,在上面的选择器中,选择 OS:Windows,Package:Pip 以及适合您机器的 CUDA 版本。通常,最新的 CUDA 版本更好。然后,运行显示给您的命令。

    验证

    为了确保 PyTorch 已正确安装,我们可以通过运行 PyTorch 示例代码来验证安装。在这里,我们将构造一个随机初始化的张量。

    在命令行中,键入

    python
    

    然后输入以下代码

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

    输出应类似于

    tensor([[0.3380, 0.3845, 0.3217],
            [0.8337, 0.9050, 0.2650],
            [0.2979, 0.7141, 0.9069],
            [0.1449, 0.1132, 0.1375],
            [0.4675, 0.3947, 0.1426]])
    

    此外,要检查您的 GPU 驱动程序和 CUDA 是否已启用并可供 PyTorch 访问,请运行以下命令以返回 CUDA 驱动程序是否已启用

    import torch
    torch.cuda.is_available()
    

    从源代码构建

    对于大多数 PyTorch 用户而言,通过包管理器从预构建的二进制文件安装将提供最佳体验。但是,有时您可能希望安装最新的 PyTorch 代码,无论是用于测试还是在 PyTorch 核心上进行实际开发。要安装最新的 PyTorch 代码,您需要从源代码构建 PyTorch

    先决条件

    1. 安装 Anaconda
    2. 如果您的机器具有 支持 CUDA 的 GPU,请安装 CUDA
    3. 如果您想在 Windows 上构建,还需要 Visual Studio 以及 MSVC 工具集和 NVTX。这些依赖项的确切要求可以在 此处 找到。
    4. 请按照此处描述的步骤操作:https://github.com/pytorch/pytorch#from-source

    您可以按照 上面 所述验证安装。