-
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 댓글