search

deeplearningzerotoall / PyTorch

Deep Learning Zero to All - Pytorch
1.03k stars 1.33k forks source link

GPU / CPU 코드 작성 가이드 #2

Closed simonjisu closed 5 years ago

simonjisu commented 5 years ago 
import torch
device = 'gpu' if torch.cuda.is_available() else 'cpu' 
x =torch.rand(3).to(device)
x.device

output:

device(type='cpu')

이런 식으로 하는게 좋을까요? 아니면 조금더 깔끔하게 cpu 사용을 default 로 하고, gpu 사용할 것만 to(device) 로 전달 할까요?

simonjisu commented 5 years ago

'gpu' 가 아니라 'cuda' 에요. 오타입니다