윈도우 10(windows 10)에서 PyTorch 설치하고 gpu를 사용하자!(PyTorch install)

요즘 Keras를 사용하다가 PyTorch를 공부하고 있습니다.

PyTorch는 또 다른 딥러닝 라이브러리이죠

윈도우에서 지원이 미약하다가 이제는 좀 잘 되는 것 같은데요 ㅎㅎ

PyTorch가 version이 up 되면서 1.1.0 버전이 나왔습니다.

이번 포스팅은 PyTorch를 윈도우 10에서 (windows 10) 설치하는 방법에 대해서 알아봅니다.

그리고 GPU도 실행시키구요

저는 CUDA 버전이 9.0인데요

 

먼저 윈도우에서 cuda 버전을 확인하는 방법부터 보죠!

 

cuda version check in windows

 

바로 cmd를 켜서 nvcc --version을 하시면 cuda version을 볼 수 있습니다.

저는 release 9.0이라고 나와있네요!

이렇게 cuda 버전을 확인했으면 이제 pytorch를 설치해볼까요?

 

윈도우에서는 anaconda로 설치하셔야합니다.

 

 

 

여기서 conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

를 입력을 합니다.

torchvision은 pytorch에서 image 처리를 할 때 사용합니다. 같이 설치해주시면 좋습니다.

저렇게 하면 windows에서 pytorch 설치를 하실 수 있습니다!

 

 

이런 화면이 쭉쭉 나오면서 진행됩니다

y도 중간에 입력해주시구요 ㅎㅎ

 

설치가 끝났으면 이제 파이토치가 gpu 지원이 되는지 확인해야겠죠?

import torch

torch.cuda.get_device_name(0)

torch.duda.is_available()

을 통해 True가 보이시고 장비가 보이시면 연동이 된 것입니다!

 

그리고 torch.__version__을 입력하면 버전이 나오구요!

저는 1.1.0이 나왔네요!

 

여기까지 windows10 pytorch 설치였습니다~