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()