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