PyTorch 를 통해 GPU를 사용하는 것은 매우 쉽습니다. 먼저, 모델을 GPU에 넣습니다: device = torch.device("cuda:0") model.to(device) Copy to clipboard.
確定! 回上一頁