Tag: pytorch

PyTorch GPU setting

PyTorch GPU 버전을 설치한다고 GPU를 사용하지 않습니다. GPU가 여럿이라면 어떤 GPU가 사용 가능한지 확인하고, 변수와 모델에 GPU 사용 설정을 해야 합니다. 1. PyTorch 버전 확인 현재 버전을 확인합니다.123import torch, torchvisionprint(torch.__version__)print(torchvision.