파이토치

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

버전 확인 import torch print(torch.__version__) GPU 연결 확인 device = 'cuda' if torch.cuda.is_available() else 'cpu' print(device)