윈도우10(windows 10)에서 Pytorch 설치하고 GPU사용하기
본문 바로가기
Python_basic

윈도우10(windows 10)에서 Pytorch 설치하고 GPU사용하기

by Migos 2019. 10. 3.
반응형

대학원을 와서 제일 잘 한 것중 하나인 그래픽카드 구매. 물론 학교돈이긴 하지만.. RTX2080을 예전쓰던 컴에 꾸역꾸역 박아넣었다...

이제 먼지묵은 그래픽카드를 돌려보려고 파이토치를 공부중에 있다. 

오늘은 오전에 python kernel 만지작거리다가 지워버려서.. 오전을 날려버리고 결국 아나콘다 재설치를 감행하여 해결하였다. 

 

그래서 잘 쓰고있던 pytorch도 재설치, torchvision도 재설치.. 뭔가 초심으로 돌아간 마음으로 하나하나 다시 설치중이다..ㅠㅠ

 

나중에 또 설치하게되면(그럼안되지만..) 참고하기위해 글을 남겨본다.

 

우선.. 나는 당연히 pip install pytorch 를 터미널에서 치고 오류가 나서 검색해보던 중 여윽시 미리 올려주신 블로그가 있어 참고할 수 있었다.(https://lsjsj92.tistory.com/494)

오류화면

1. 우선 맨처음 내 컴터에 CUDA 가 버젼이 몇으로 깔려있는지 확인해야한다.

  1. cmd 터미널을 켠다(Anaconda Prompt 커맨드는 마우스 우클릭-관리자권한으로실행 으로 켠다).
  2. nvcc --version

   3. conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

잘 깔렸는지 확인해보자.

오류화면을 캡쳐를 못했으나.. permission error 비스무리한게 뜨면서 설치가 안되었다.

이럴땐 아나콘다 프롬프트 오른쪽 클릭하여 권한으로 실행을 꼭!! 눌러서 터미널을 열어준다.

 

그다음 안전하게 pytorch 사이트 들어가서 아래와 같이 내 환경에 맞는것을 클릭 해주고

Run this Command: 를 긁어서 복사붙여넣기 하기만 하면 잘 설치된다.

 

 

이제 잘 깔렸는지 확인하는 단계. 아래와 같이 나오면 정상!! 

python
import torch
torch.cuda.get_device_name(0)
torch.cuda.is_available()
print(torch.__version__)

끝.

반응형

댓글