GPU 설정

torch.manual_seed(RANDOM_STATE)

def get_device() -> torch.device:
    if torch.cuda.is_available():
        device = torch.device("cuda")
    elif torch.backends.mps.is_available():
        device = torch.device("mps")
    else:
        device = torch.device("cpu")

    print(f"🚀 사용 중인 디바이스: {device}")
    return device

device = get_device()