[Pytorch] Pytorch 버전 확인과 GPU 연결 확인

버전 확인

import torch

print(torch.__version__)

---

GPU 연결 확인

device = 'cuda' if torch.cuda.is_available() else 'cpu'

print(device)