Ordinary existence

engineer 감각이 있는 Data researcher

sun about me

Programming/python

window PyTorch 설치하기

Mithumbsup 2021. 9. 2. 16:55

PyTorch 

 - 페이스북 인공지능 연구팀에서 개발한 파이썬(Python) 프로그래밍 언어 기반의 딥러닝 프레임워크

 

[ 출처 : PyTorch 공식 홈페이지 설치방법 ]

 - Start Locally Documentaiton 링크 : [ https://pytorch.org/get-started/locally/ ]

 

PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

pytorch.org

지금 내 pc 환경은 window에서  pip 환경으로 구성되어져 있음으로,

pip 환경에서 pytorch를 설정방법으로 정리하였다!

 

<pc 환경>

- os : window 10

- nvidia : GTX 2070 SUPER

- python : 3.7.9

- pip 

 

--------------------------------------------------------------------------------------------------------------------------------

 

0. Pytorch를 설치하기 위한 환경 점검

 

    PyTorch는 Python 언어로 구성된 딥러닝 프레임워크임으로,

    PyTorch 공식 홈페이지에서는

    os별로 설치한 Python 패키지 관리도구(anaconda | pip)를 기점으로

    pc환경에서 CuDa 사용여부 선택에 따라

    설치 명령문을 각각 다르게 주어짐으로 

    아래의 3단계 확인사항을 검토해야한다.

 

     1.  Python 패키지 설정 환경 점검

         - conda or pip 

 

     2.  PyTorch를 cpu 버전 or gpu 버전으로 설치할지 확인

          2.1. cpu 버전 구성

          2.2. gpu로 구성한다면 CUDA, CuDNN 버전 설정을 확인

     

     3.  선택한  각 cpu 버전 or gpu 버전환경에 맞는 Pytorch 가 잘 설치되었는지 shell 명령문으로 확인

 

-----------------------------------------------------------------------------------------------------------------------.

 

 

1. Python 설치 환경 점검 

     - shell에서 python version을 확인한다.

python --version

 

 

2. (핵심) PyTorch를 cpu 버전 vs gpu 버전으로 설치할지 확인     

       PyTorch 공식 홈페이지에서는  아래의 pc 설치확인 사항을 선택하면, 

       Run this Command를 통해 내 pc 환경에 맞는 PyTorch 설치 명령문을 안내해준다~

      따라서 2-1) cpu버전 Pytorch   2-2) gpu버전 Pytorch  로 나누어 정리!

<Pytorch 설치를 위한  pc환경 검토사항>

 

 

2-1) cpu버전 Pytorch 설치 명령문 확인

    

   1.  위에 첨부한 공식 홈페이지에 방문해서 위에서 확인한 pc 환경을 선택한 후 

        cpu 버전 명령문을 확인하면 ok

conda install pytorch torchvision torchaudio cpuonly -c pytorch

 

  2. Pytorch cpu 설치 확인 명령문

   본인 pc 환경에 Pytorch cpu 가 잘 설치되었는지 점검하기 위한 명령문!

 

python 

>>>import torch
>>>torch.FloatTensor(10)
tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], device='cuda:0')

 

 

 

2-2) gpu버전 Pytorch 설치 명령문

      

 좀 복잡쓰하다

 먼저 gpu 버전으로 Pytorch를 설치했을 때 사용가능한 여부 확인을 위해 

 

 

0. 그래픽카드 장치 작동 확인

   shell에서 아래의 명령어를 통해서 pc에 설치되어있는 그래픽카드 장치 작동을 확인한다.

nvidia-smi

   

 

0.1. 작동이 된다면 shell에서 CUDA, CuDNN 버전을 확인할 수 있다. 

    

  0.2. 작동이 되지않는다면 gpu버전의 PyTorch를 설치하기 위해

        각자의 os 맞는 그래픽카드장치가 있는지 확인해보고,

       그래픽카드 성능에 호환되는 CUDA, CuDNN 버전을 설치하는 과정을 찾아봐야할 듯하다!

 

        CUDA 설치 확인 방법 정리 : https://mithumbsup.tistory.com/5 

 

 

1. PyTorch gpu 버전 설치 명령문 확인

    위에 첨부한 공식 홈페이지에 방문해서 위에서 확인한 pc 환경을 선택한 후 

    gpu 버전 명령문을 확인하면 ok

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

 

 

2. Pytorch gpu 설치 확인 명령문

   본인 pc 환경에 Pytorch gpu 가 잘 설치되었는지 점검하기 위한 명령문!

 

python 

>>>import torch
>>>torch.cuda.FloatTensor(10)
tensor([0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], device='cuda:0')

 

 

 

 

 

ok