import os
import torch

def setup_environment():
    """
    OCR 모델 실행에 필요한 환경 변수를 설정합니다.
    """
    if torch.version.cuda == "11.8":
        os.environ["TRITON_PTXAS_PATH"] = "/usr/local/cuda-11.8/bin/ptxas"

    os.environ["VLLM_USE_V1"] = "0"
    os.environ["CUDA_VISIBLE_DEVICES"] = "0"