PyTorch 설치 및 구성

이 자습서의 이전 단계에서는 PyTorch의 기본 사항과 이를 사용하여 기계 학습 모델을 만드는 데 필요한 필수 구성 요소에 대해 설명했습니다. 여기서는 컴퓨터에 설치합니다.

PyTorch 가져오기

이 자습서에서는 64비트 Windows Anaconda 환경을 사용하여 CPU에서 모델을 학습하고 실행합니다. 자습서의 뒷부분에서 Windows ML 애플리케이션의 경우 Windows 10 버전 1809(빌드 17763) 이상이 필요합니다. 이러한 단계에는 GPU 또는 CUDA가 필요하지 않습니다.

중요합니다

이 자습서는 레거시 자습서입니다. 이 단계는 다운스트림 학습 및 ONNX 내보내기 코드를 업그레이드하는 대신 원래 설치 예제에 표시된 Python 3.8 및 PyTorch 1.7.1 패키지 버전을 유지합니다. Python 3.8은 더 이상 지원되지 않으며 PyTorch는 버전 2.6부터 공식 Conda 패키지 게시를 중지했습니다. 별도의 테스트 머신 또는 가상 머신을 사용하여 프로덕션 환경이 아닌 이 환경을 재현합니다. 새 프로젝트의 경우 현재 PyTorch 설치 지침을 따르세요. 이 자습서에서는 최신 패키지의 유효성을 검사하지 않습니다.

  1. Anaconda 설치 관리자 Anaconda3-2020.11-Windows-x86_64.exe 보관 파일에서 64비트 Windows 설치 관리자를 다운로드하여 테스트 머신에 설치합니다. Anaconda 2020.11에는 Python 3.8.5 및 자습서의 뒷부분에서 사용되는 데이터 분석 패키지가 포함되어 있습니다.

    Python 3.8 Windows 설치 프로그램을 보여주는 보관된 Anaconda 다운로드 페이지입니다.

  2. Anaconda3 > Anaconda PowerShell 프롬프트 시작을 >엽니다. Python 및 Conda 버전을 확인합니다.

    python --version
    conda --version
    

    계속하기 전에 Python 버전 3.8을 보고했는지 확인합니다.

    Python 및 Conda 버전을 표시하는 Anaconda PowerShell 프롬프트입니다.

  3. 이전 PyTorch 버전 페이지에서v1.7.1 > Conda > Linux 및 Windows > CPU 전용 지침을 검토합니다. 이 자습서에서는 안정적인 현재 릴리스 선택기가 아닌 Windows, Conda, Python 및 CPU 옵션을 사용합니다. CUDA 빌드는 호환되는 NVIDIA GPU에 사용할 수 있지만 GPU 설정은 이 절차의 일부가 아닙니다.

    Windows, Conda, Python 및 CUDA 없음을 보여 주는 이전 PyTorch 1.7.1 선택기입니다.

  4. 동일한 Anaconda PowerShell 프롬프트에서 CPU 패키지를 설치합니다. 최신 PyTorch 릴리스를 선택하지 않도록 버전이 고정됩니다.

    conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cpuonly -c pytorch
    

    PyTorch 1.7.1 및 해당 보조 패키지의 CPU 빌드를 나열하는 Anaconda 패키지 계획입니다.

    스크린샷은 기록 예제입니다. Python 버전과 고정되지 않은 명령은 다를 수 있습니다. 텍스트의 명령을 따릅니다.

  5. 패키지 계획을 검토하고, 메시지가 표시되면 입력 y 하고, 설치가 완료될 때까지 기다립니다. Conda가 레거시 패키지를 확인할 수 없는 경우 최신 PyTorch 릴리스를 대체하지 말고 이후 자습서 단계가 호환된다고 가정합니다.

  6. 동일한 프롬프트에서 Python 시작합니다.

    python
    

    다음 코드를 입력하여 임의로 초기화된 텐서를 생성합니다.

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

    출력은 임의의 2x3 텐서(행 2개와 열 3개)여야 합니다. 숫자는 다르지만 셰이프는 다음 예제와 일치해야 합니다.

    tensor([[0.5343, 0.7362, 0.8390],
            [0.4758, 0.2399, 0.1936]])
    

    Python 두 개의 행과 세 개의 열이 있는 텐서를 인쇄합니다.

    Anaconda PowerShell 프롬프트로 돌아가려면 입력 exit() 합니다. 자습서의 다음 단계에서 Python 인터프리터를 선택할 때 동일한 Anaconda 환경을 사용합니다.

비고

더 알아보고 싶으신가요? PyTorch 공식 웹 사이트를 방문하십시오.

다음 단계

이제 PyTorch를 설치했으므로 모델에 대한 데이터를 설정할 준비가 되었습니다.