[개발 환경] 윈도우에 CUDA 버전에 맞게 torch 설치, pytorch GPU 사용

ahnty0122.tistory.com/37?category=454641

[환경설정] 윈도우(Windows)에 Tensorflow-gpu 설치(NVIDIA driver, CUDA Toolkit, cuDNN 설치)

앞선 글에 적어놓았던 nvidia driver 설치, cuda 설치, cudnn 설치를 완료한 후 진행하면 torch 설치는 아주 쉽다.

---

우선 cmd 켜서 cuda 버전 확인

nvcc --version

cuda가 10.0이므로 그에 맞는 torch를 설치해주면 된다.

 

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

입력후 엔터 누르면 torch 설치 완료!

---

그 후 torch에서 gpu 사용 가능한지 확인

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

gpu 드라이버 이름 나오고, torch cuda available도 True로 확인 완료 ㅎㅎ