13 lines
328 B
Python
13 lines
328 B
Python
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"
|