ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • Windows Pytorch 설치
    etc 2018. 3. 2. 23:50
    반응형
    CUDA는 9.0 버전으로 한다.


    1. Anaconda 3 설치


       다운로드 : https://anaconda.org/


       Miniconda를 설치하면 안된다. 반드시 Anaconda를 설치해야 한다.


       설치를 마치고, 폴더를 보면 scripts가 보이는데, 이놈을 PATH에 걸어준다.


    2. 가상환경 설치

       텐서플로우와 Pytorch를 서로다른 환경에서 함께 사용하기 위해서 설정한다.


       1) 가상환경 생성 / 지우기

          > conda create --name torch

          > conda remove --name torch --all

        

       2) 가상환경 활성화 / 비활성화

          > activate torch

          > deactivate


    3. pytorch 설치


       > conda install -c peterjc123 pytorch cuda90



    4. 테스트

       python으로 들어가서...


       import torch


       해서 오류가 없으면 성공이다.


    5. 몇가지 모듈 설치 방법

       아래 모듈은 기본으로 포함되어 있지 않으므로 설치해야 한다.

       1) openCV

        

            > conda install -c conda-forge opencv 


       2) torchvision


          소스 다운로드 : https://github.com/pytorch/vision 


           > python setup.py install


       3) pillow-simd

        

          이미지 처리속도를 높여준다. 기존에 pillow가 설치되어 있으면 아래처럼 uninstall 시켜준다.

             > conda uninstall pillow

          

          윈도우 전용을 다운로드 한다.

             링크 : https://www.lfd.uci.edu/~gohlke/pythonlibs/#pillow-simd


          설치

             pip install [위에서 다운로드 받은 파일명]

    끝.
    반응형

    'etc' 카테고리의 다른 글

    jekyll 테마 사이트  (0) 2018.03.10
    Visual Studio 컴파일시 결과창 바로꺼짐현상  (0) 2018.03.07
    Tensorboard 사용하기  (0) 2018.02.23
    자격증 기출문제 모음 페이지  (0) 2018.02.21
    정부 Open API 관련 사이트  (0) 2018.02.01

    댓글

Designed by Tistory.