Friday, February 9, 2018

PyTorch 설치

뭐 대부분 설치하는건 비슷하더라..

* Windows
  1. CUDA, cudnn 설치하고 anaconda 설치하고
  2. conda create -n py36torch python=3.6
  3. activate py36torch
  4. conda install -c peterjc123 pytorch cuda90
  5. 테스트
    1. import torch
    2. x = torch.rand(5,3)
    3. print(x)
    4. print(torch.cuda.is_available())

* Mac

  1. conda create -n py36torch python=3.6
  2. source activate py36torch
  3. conda install pytorch torchvision -c pytorch
  4. 테스트
    1. import torch
    2. x = torch.rand(5,3)
    3. print(x)