2020-08-12 15:02:01

저는 지금까지 Tensorflow를 주로 사용해왔는데, Pytorch도 사용해야 할 일이 생겨서 설치법을 알아보고 설치를 완료했습니다. 

 

CUDA10.0에 맞는 pytorch와 torchvision의 버전은 pytorch 1.2와 torchvision 0.4입니다. 이를 Anaconda에서 설치하려면 다음과 같은 명령어로 설치할 수 있습니다. 

 

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

 

설치가 잘 되었는지 확인하려면, 아래 코드를 실행해보시면 됩니다. 

 

1
2
import torch
torch.cuda.is_available()
cs

 

 

위와 같이 True라고 나오면 제대로 설치된 것입니다. GPU를 이용할 수 있다는 뜻으로 생각하셔도 됩니다. 

 

 

<참고자료>

[1] https://pytorch.org/get-started/previous-versions/, PyTorch, "INSTALLING PREVIOUS VERSIONS OF PYTORCH"