Author: byun1114

Python은 문자로 결과가 나오더라도 문자가 아닌 것 같은 경우가 있다. 문자인 숫자를 숫자로 인식시켜주려면 int()로 변경을 해야 하듯이 문자열 결과를 기반으로 조건을 확인하려면 문자열로 변경시켜 주어야 한다.

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

if str(device) == 'cpu':
    model = torch.load('model_class9_entire.pt', map_location='cpu')
    model = model.to(device)
    model.eval()
elif str(device) == 'cuda':
    model = torch.load('model_class9_entire.pt')
    model = model.to(device)
    model.eval()

PyTorch 1.7.0 버젼이 나왔다고 한다. 아직 버젼에 의존하는 그런 고급(?)은 아니니 최신으로 업그레이드 해보기로 했다. PIP로 패키지를 관리하는데 이 프로그램은 놀랍게도 upgrade에 대한 설명이 없다! 하지만, 해야 한다면 다음과 같은 형식으로 한다. 셀프 업그레이드 및 패키지 업그레이느는 다음과 같다.

pip3 install --upgrade pip
pip3 install --upgrade 패키지

PyTorch 버젼은 다음과 같이 확인한다. Python에서 확인해야 한다.

import torch
print(torch.__version__)

근데 PyTorch는 그렇게 하면 안되고 홈페이지에서 조건에 맞춰 다시 설치하면 된다.

pip install torch==1.7.0+cpu torchvision==0.8.1+cpu torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

GPU 에서 작업하고 저장한 모델이다. 하지만 GPU가 없는 컴퓨터에서 작업을 하기 위해서 모델을 불러오면 오류가 발생한다. 오류 메세지에 어떻게 하라는 설명이 나오기는 하니까 어렵지 않게 해결할 수 있다. map_location = ‘cpu’를 명시하면서 불러오면 된다.

model = torch.load('model.pt', map_location='cpu')