Pytorch 설치하기

 


Pytorch 설치하기

환경에 맞는 Pytorch 설치하기

image pytorch 사이트 main

Pytorch 메인 페이지에서 스크롤하면, 다음 그림과 같은 설치 옵션을 선택할 수 있는 창이 있습니다. 현재 자신의 설정에 맞게 설정을 한 후에, Run this Command 에 나온 명령어를 복사한 후에 terminal에 입력하여 pytorch를 설치하세요.

아래와 같이 진행됩니다.

(base) C:\Users\user>conda install pytorch torchvision cpuonly -c pytorch
WARNING: The conda.compat module is deprecated and will be removed in a future release.
Collecting package metadata: done
Solving environment: done

## Package Plan ##

  environment location: C:\Users\user\Anaconda3

  added / updated specs:
    - cpuonly
    - pytorch
    - torchvision


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    conda-4.7.11               |           py37_0         3.0 MB
    conda-package-handling-1.3.11|           py37_0         280 KB
    cpuonly-1.0                |                0           2 KB  pytorch
    ninja-1.9.0                |   py37h74a9793_0         263 KB
    pytorch-1.2.0              |      py3.7_cpu_1        71.5 MB  pytorch
    torchvision-0.4.0          |         py37_cpu         2.0 MB  pytorch
    ------------------------------------------------------------
                                           Total:        77.0 MB

The following NEW packages will be INSTALLED:

  conda-package-han~ pkgs/main/win-64::conda-package-handling-1.3.11-py37_0
  cpuonly            pytorch/noarch::cpuonly-1.0-0
  ninja              pkgs/main/win-64::ninja-1.9.0-py37h74a9793_0
  pytorch            pytorch/win-64::pytorch-1.2.0-py3.7_cpu_1
  torchvision        pytorch/win-64::torchvision-0.4.0-py37_cpu

The following packages will be UPDATED:

  conda                                       4.6.11-py37_0 --> 4.7.11-py37_0


Proceed ([y]/n)?


Downloading and Extracting Packages
ninja-1.9.0          | 263 KB    | ############################################################################ | 100%
conda-package-handli | 280 KB    | ############################################################################ | 100%
pytorch-1.2.0        | 71.5 MB   | ############################################################################ | 100%
conda-4.7.11         | 3.0 MB    | ############################################################################ | 100%
torchvision-0.4.0    | 2.0 MB    | ############################################################################ | 100%
cpuonly-1.0          | 2 KB      | ############################################################################ | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done

torchvision torchtext 설치하기

torchvision은 이미지 처리용 데이터셋 처리를, torchtext는 글과 같은 데이터셋 처리를 도와주는 도구입니다. 아래의 명령어를 입력하여 설치해줍니다.

pip3 install torchvision torchtext

위와 같은 요소들을 다 설치하면 대략적인 설치는 끝납니다.


Pytorch를 이용한 예제 수행

tensor 출력하기

기본적인 tensor를 출력해보는 예제입니다. tensor는 쉽게 설명하면 행렬과 같은 개념이라고 보시면 됩니다. 아래와 같은 python 코드가 필요합니다.

import torch

x = torch.tensor([[1,2,3], [4,5,6], [7,8,9]])
print(x)

해당 소스코드를 python 명령어를 이용하여 윈도우 console 창인 cmd에서 수행하면 아래와 같은 결과를 얻을 수 있습니다.

C:\Users\user> python pytorch_ch3.py
tensor([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])

위와 같이 결과가 나오면 성공입니다! 향후 더 나은 예제들을 Pytorch를 이용하여 수행할 수 있습니다.