jeonghyo.k
3946ff6a25
- prediction/image/ FastAPI 서버 Docker 환경 구성 - Dockerfile: PyTorch 2.1 + CUDA 12.1 기반 GPU 이미지 - docker-compose.yml: GPU 할당 + 데이터 볼륨 마운트 - requirements.txt: 서버 의존성 목록 - .env.example: 환경변수 템플릿 - DOCKER_USAGE.md: 빌드/실행/API 사용법 문서 - Dockerfile에 .dockerignore 제외 폴더 mkdir -p 추가 - .gitignore: prediction/image 결과물 및 모델 가중치(.pth) 제외 추가 - dbInsert_csv.py, dbInsert_shp.py 삭제 (미사용 DB 로직) - api.py: dbInsert import 및 주석 처리된 DB 호출 코드 제거 - aerialRouter.ts: req.params 타입 오류 수정
56 lines
2.3 KiB
Python
56 lines
2.3 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import os
|
|
import platform
|
|
|
|
import cv2
|
|
import torch.multiprocessing as mp
|
|
|
|
from ..utils import get_root_logger
|
|
|
|
|
|
def setup_multi_processes(cfg):
|
|
"""Setup multi-processing environment variables."""
|
|
logger = get_root_logger()
|
|
|
|
# set multi-process start method
|
|
if platform.system() != 'Windows':
|
|
mp_start_method = cfg.get('mp_start_method', None)
|
|
current_method = mp.get_start_method(allow_none=True)
|
|
if mp_start_method in ('fork', 'spawn', 'forkserver'):
|
|
logger.info(
|
|
f'Multi-processing start method `{mp_start_method}` is '
|
|
f'different from the previous setting `{current_method}`.'
|
|
f'It will be force set to `{mp_start_method}`.')
|
|
mp.set_start_method(mp_start_method, force=True)
|
|
else:
|
|
logger.info(
|
|
f'Multi-processing start method is `{mp_start_method}`')
|
|
|
|
# disable opencv multithreading to avoid system being overloaded
|
|
opencv_num_threads = cfg.get('opencv_num_threads', None)
|
|
if isinstance(opencv_num_threads, int):
|
|
logger.info(f'OpenCV num_threads is `{opencv_num_threads}`')
|
|
cv2.setNumThreads(opencv_num_threads)
|
|
else:
|
|
logger.info(f'OpenCV num_threads is `{cv2.getNumThreads()}')
|
|
|
|
if cfg.data.workers_per_gpu > 1:
|
|
# setup OMP threads
|
|
# This code is referred from https://github.com/pytorch/pytorch/blob/master/torch/distributed/run.py # noqa
|
|
omp_num_threads = cfg.get('omp_num_threads', None)
|
|
if 'OMP_NUM_THREADS' not in os.environ:
|
|
if isinstance(omp_num_threads, int):
|
|
logger.info(f'OMP num threads is {omp_num_threads}')
|
|
os.environ['OMP_NUM_THREADS'] = str(omp_num_threads)
|
|
else:
|
|
logger.info(f'OMP num threads is {os.environ["OMP_NUM_THREADS"] }')
|
|
|
|
# setup MKL threads
|
|
if 'MKL_NUM_THREADS' not in os.environ:
|
|
mkl_num_threads = cfg.get('mkl_num_threads', None)
|
|
if isinstance(mkl_num_threads, int):
|
|
logger.info(f'MKL num threads is {mkl_num_threads}')
|
|
os.environ['MKL_NUM_THREADS'] = str(mkl_num_threads)
|
|
else:
|
|
logger.info(f'MKL num threads is {os.environ["MKL_NUM_THREADS"]}')
|