Shortcuts

    직접 설치하기

    사용 환경을 선택하고 설치 명령을 복사해서 실행해 보세요. Stable 버전은 테스트 및 지원되고 있는 가장 최근의 PyTorch 버전으로, 대부분의 사용자에게 적합합니다. Preview 버전은 아직 완전히 테스트되지 않은 최신 1.10 버전으로 매일 업데이트됩니다. 사용 중인 패키지 매니저에 따라 아래의 사전 요구사항(예: numpy)이 충족되었는지 확인해 주세요. 모든 의존성을 설치할 수 있는 Anaconda를 패키지 매니저로 추천합니다. 이전 버전의 PyTorch도 설치할 수 있습니다. LibTorch는 C++에서만 지원합니다.

    PyTorch 빌드
    OS 종류
    패키지 매니저
    언어
    플랫폼
    이 명령을 실행하세요:
    PyTorch 빌드
    Stable (1.9.0)
    Preview (Nightly)
    LTS (1.8.1)
    OS 종류
    Linux
    Mac
    Windows
    패키지 매니저
    Conda
    Pip
    LibTorch
    Source
    언어
    Python
    C++ / Java
    플랫폼
    CUDA 10.2
    CUDA 11.1
    ROCm 4.2 (beta)
    CPU
    이 명령을 실행하세요:
    conda install pytorch torchvision -c pytorch

    Installing on macOS

    PyTorch can be installed and used on macOS. Depending on your system and compute requirements, your experience with PyTorch on a Mac may vary in terms of processing time. It is recommended, but not required, that your Mac have an NVIDIA GPU in order to harness the full power of PyTorch’s CUDA support.

    Currently, CUDA support on macOS is only available by building PyTorch from source

    Prerequisites

    macOS Version

    PyTorch is supported on macOS 10.10 (Yosemite) or above.

    Python

    It is recommended that you use Python 3.5 or greater, which can be installed either through the Anaconda package manager (see below), Homebrew, or the Python website.

    Package Manager

    To install the PyTorch binaries, you will need to use one of two supported package managers: Anaconda or pip. Anaconda is the recommended package manager as it will provide you all of the PyTorch dependencies in one, sandboxed install, including Python.

    Anaconda

    To install Anaconda, you can download graphical installer or use the command-line installer. If you use the command-line installer, you can right-click on the installer link, select Copy Link Address, and then use the following commands:

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

    pip

    Python 3

    If you installed Python via Homebrew or the Python website, pip was installed with it. If you installed Python 3.x, then you will be using the command pip3.

    Tip: If you want to use just the command pip, instead of pip3, you can symlink pip to the pip3 binary.

    Installation

    Anaconda

    To install PyTorch via Anaconda, use the following conda command:

    conda install pytorch torchvision -c pytorch
    

    pip

    To install PyTorch via pip, use one of the following two commands, depending on your Python version:

    # Python 3.x
    pip3 install torch torchvision
    

    Verification

    To ensure that PyTorch was installed correctly, we can verify the installation by running sample PyTorch code. Here we will construct a randomly initialized tensor.

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

    The output should be something similar to:

    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]])
    

    Building from source

    For the majority of PyTorch users, installing from a pre-built binary via a package manager will provide the best experience. However, there are times when you may want to install the bleeding edge PyTorch code, whether for testing or actual development on the PyTorch core. To install the latest PyTorch code, you will need to build PyTorch from source.

    You will also need to build from source if you want CUDA support.

    Prerequisites

    1. Install Anaconda
    2. Install CUDA, if your machine has a CUDA-enabled GPU.
    3. Follow the steps described here: https://github.com/pytorch/pytorch#from-source

    You can verify the installation as described above.

    Installing on Linux

    PyTorch can be installed and used on various Linux distributions. Depending on your system and compute requirements, your experience with PyTorch on Linux may vary in terms of processing time. It is recommended, but not required, that your Linux system has an NVIDIA GPU in order to harness the full power of PyTorch’s CUDA support..

    Prerequisites

    Supported Linux Distributions

    PyTorch is supported on Linux distributions that use glibc >= v2.17, which include the following:

    The install instructions here will generally apply to all supported Linux distributions. An example difference is that your distribution may support yum instead of apt. The specific examples shown were run on an Ubuntu 18.04 machine.

    Python

    Python 3.6 or greater is generally installed by default on any of our supported Linux distributions, which meets our recommendation.

    Tip: By default, you will have to use the command python3 to run Python. If you want to use just the command python, instead of python3, you can symlink python to the python3 binary.

    However, if you want to install another version, there are multiple ways:

    If you decide to use APT, you can run the following command to install it:

    sudo apt install python
    

    It is recommended that you use Python 3.6, 3.7 or 3.8, which can be installed via any of the mechanisms above .

    If you use Anaconda to install PyTorch, it will install a sandboxed version of Python that will be used for running PyTorch applications.

    Package Manager

    To install the PyTorch binaries, you will need to use one of two supported package managers: Anaconda or pip. Anaconda is the recommended package manager as it will provide you all of the PyTorch dependencies in one, sandboxed install, including Python.

    Anaconda

    To install Anaconda, you will use the command-line installer. Right-click on the 64-bit installer link, select Copy Link Location, and then use the following commands:

    # 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.`
    

    You may have to open a new terminal or re-source your ~/.bashrc to get access to the conda command.

    pip

    Python 3

    While Python 3.x is installed by default on Linux, pip is not installed by default.

    sudo apt install python3-pip
    

    Tip: If you want to use just the command pip, instead of pip3, you can symlink pip to the pip3 binary.

    Installation

    Anaconda

    No CUDA

    To install PyTorch via Anaconda, and do not have a CUDA-capable system or do not require CUDA, in the above selector, choose OS: Linux, Package: Conda and CUDA: None. Then, run the command that is presented to you.

    With CUDA

    To install PyTorch via Anaconda, and you do have a CUDA-capable system, in the above selector, choose OS: Linux, Package: Conda and the CUDA version suited to your machine. Often, the latest CUDA version is better. Then, run the command that is presented to you.

    pip

    No CUDA

    To install PyTorch via pip, and do not have a CUDA-capable system or do not require CUDA, in the above selector, choose OS: Linux, Package: Pip and CUDA: None. Then, run the command that is presented to you.

    With CUDA

    To install PyTorch via pip, and do have a CUDA-capable system, in the above selector, choose OS: Linux, Package: Pip and the CUDA version suited to your machine. Often, the latest CUDA version is better. Then, run the command that is presented to you.

    Verification

    To ensure that PyTorch was installed correctly, we can verify the installation by running sample PyTorch code. Here we will construct a randomly initialized tensor.

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

    The output should be something similar to:

    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]])
    

    Additionally, to check if your GPU driver and CUDA is enabled and accessible by PyTorch, run the following commands to return whether or not the CUDA driver is enabled:

    import torch
    torch.cuda.is_available()
    

    Building from source

    For the majority of PyTorch users, installing from a pre-built binary via a package manager will provide the best experience. However, there are times when you may want to install the bleeding edge PyTorch code, whether for testing or actual development on the PyTorch core. To install the latest PyTorch code, you will need to build PyTorch from source.

    Prerequisites

    1. Install Anaconda
    2. Install CUDA, if your machine has a CUDA-enabled GPU.
    3. Follow the steps described here: https://github.com/pytorch/pytorch#from-source

    You can verify the installation as described above.

    Installing on Windows

    PyTorch can be installed and used on various Windows distributions. Depending on your system and compute requirements, your experience with PyTorch on Windows may vary in terms of processing time. It is recommended, but not required, that your Windows system has an NVIDIA GPU in order to harness the full power of PyTorch’s CUDA support.

    Prerequisites

    Supported Windows Distributions

    PyTorch is supported on the following Windows distributions:

    The install instructions here will generally apply to all supported Windows distributions. The specific examples shown will be run on a Windows 10 Enterprise machine

    Python

    Currently, PyTorch on Windows only supports Python 3.x; Python 2.x is not supported.

    As it is not installed by default on Windows, there are multiple ways to install Python:

    If you use Anaconda to install PyTorch, it will install a sandboxed version of Python that will be used for running PyTorch applications.

    If you decide to use Chocolatey, and haven’t installed Chocolatey yet, ensure that you are running your command prompt as an administrator.

    For a Chocolatey-based install, run the following command in an administrative command prompt:

    choco install python
    

    Package Manager

    To install the PyTorch binaries, you will need to use at least one of two supported package managers: Anaconda and pip. Anaconda is the recommended package manager as it will provide you all of the PyTorch dependencies in one, sandboxed install, including Python and pip.

    Anaconda

    To install Anaconda, you will use the 64-bit graphical installer for PyTorch 3.x. Click on the installer link and select Run. Anaconda will download and the installer prompt will be presented to you. The default options are generally sane.

    pip

    If you installed Python by any of the recommended ways above, pip will have already been installed for you.

    Installation

    Anaconda

    To install PyTorch with Anaconda, you will need to open an Anaconda prompt via Start | Anaconda3 | Anaconda Prompt.

    No CUDA

    To install PyTorch via Anaconda, and do not have a CUDA-capable system or do not require CUDA, in the above selector, choose OS: Windows, Package: Conda and CUDA: None. Then, run the command that is presented to you.

    With CUDA

    To install PyTorch via Anaconda, and you do have a CUDA-capable system, in the above selector, choose OS: Windows, Package: Conda and the CUDA version suited to your machine. Often, the latest CUDA version is better. Then, run the command that is presented to you.

    pip

    No CUDA

    To install PyTorch via pip, and do not have a CUDA-capable system or do not require CUDA, in the above selector, choose OS: Windows, Package: Pip and CUDA: None. Then, run the command that is presented to you.

    With CUDA

    To install PyTorch via pip, and do have a CUDA-capable system, in the above selector, choose OS: Windows, Package: Pip and the CUDA version suited to your machine. Often, the latest CUDA version is better. Then, run the command that is presented to you.

    Verification

    To ensure that PyTorch was installed correctly, we can verify the installation by running sample PyTorch code. Here we will construct a randomly initialized tensor.

    From the command line, type:

    python
    

    then enter the following code:

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

    The output should be something similar to:

    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]])
    

    Additionally, to check if your GPU driver and CUDA is enabled and accessible by PyTorch, run the following commands to return whether or not the CUDA driver is enabled:

    import torch
    torch.cuda.is_available()
    

    Building from source

    For the majority of PyTorch users, installing from a pre-built binary via a package manager will provide the best experience. However, there are times when you may want to install the bleeding edge PyTorch code, whether for testing or actual development on the PyTorch core. To install the latest PyTorch code, you will need to build PyTorch from source.

    Prerequisites

    1. Install Anaconda
    2. Install CUDA, if your machine has a CUDA-enabled GPU.
    3. If you want to build on Windows, Visual Studio with MSVC toolset, and NVTX are also needed. The exact requirements of those dependencies could be found out here.
    4. Follow the steps described here: https://github.com/pytorch/pytorch#from-source

    You can verify the installation as described above.