설치 방법은 텐서플로우와 매우 유사합니다. 먼저 CUDA, cuDNN은 텐서플로우에서 설치했던것과 동일하기 때문에 링크로 들어가서 다운받아주세요~
Pytorch사이트에 들어가면 본인의 CUDA버전에 맞게 다운받는 방법을 알려줍니다!
저는 CUDA 9.0을 다운받았기 때문에 아래와 같이 선택하였습니다.(파이토치 버전은 0.4.1입니다.)
conda install pytorch torchvision -c pytorch
위의 명령어를 입력하시면 Pytorch가 설치됩니다.(시간이 오래 걸릴 수 있습니다!)
설치 확인을 위해 아래 명령어로 버전 체크를 해주세요.
python -c 'import torch; print(torch.__version__)'