[Pytorch] Pytorch Window 설치 (CUDA)

window에 CUDA와 CuDNN 설정되어 있다고 가정.
​
* 설치
1. window의 CUDA 버전확인
- window CMD 창에서 입력.

nvcc --version

 

- 명령어를 입력하면 9점대 버전인 것을 확인할 수 있다.

CMD 결과

2. 해당 버전에 맞는 Pytorch 설치

- pytorch는 anaconda에서 설치하는 것이 편하다.
- anaconda prompt 창에서 입력. (설치하고 싶은 가상환경 activate후 입력)

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

 

 

* 설치 확인

torch를 불러오고 device_name을 확인한다.

import torch
torch.cuda.get_device_name(0)