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 로 나누어 정리!
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
'Programming > python' 카테고리의 다른 글
| python Error:Microsoft Visual C++ 14.0 is required. Get it with "Build Tools for Visual Studio" 해결하기 (0) | 2021.05.12 |
|---|---|
| python - TypeError: unhashable type: 'list' (0) | 2021.03.16 |