윈도우즈에서 torchvision 깔기(with 파이토치(pytorch)깔기)

윈도우즈에서 torchvision 깔기(with 파이토치(pytorch)깔기)

윈도우즈에서 파이토치(pytorch) 깔기는 굉장히 쉽다.

아나콘다 받아서 설치하고 -> 아나콘다 프롬프트를 실행시키고 ->

(엔비디아(nvidia) 계열 그래픽카드(gpu) 사용 시) conda install -c peterjc123 pytorch

(그 외 그래픽카드 혹은 cpu만으로 사용 시) conda install -c peterjc123 pytorch-cpu

끝!

확인할 수 있는 방법은 jupytor notebook을 켜거나 그냥 아나콘다 프롬프트에서 python입력해서 python 인터프리터 창으로 들어가서

import torch

실행해서 에러가 없으면 설치 성공!

근데 문제는 딥러닝을 조금 더 공부해서 cnn(convolutional neural network)으로 넘어가게 되면 torchvision이 필요하게 되는데, 이게 설치하는 방법이 구글링을 해봐도 잘 안나온다.

바로 아나콘다 클라우드에 

conda install -c anaconda torchvision

이라는 명령어가 나오기는 하는데, 윈도우즈에서 돌리면 채널이 없다고 나온다!

홈페이지 들어가 보았더니 osx와 linux만 지원한다고 한다.

그러면 윈도우즈에서는 깔 수가 없는 것인가?

아니다!

pip 명령어를 사용하면 아주 쉽게 깔 수 있다.

pip install torchvision

끝.

위와 같이 import torchvision 해서 에러가 안나면 제대로 깔렸다고 볼 수 있다.

만약 pip에서 에러가 나면 이전 포스팅인 pip에러 수정하는 방법에 대해서 확인해보자!