# ============================
# requirements_x86.txt (Linux x86_64, CUDA 11.8)
# ============================

# PyTorch CUDA 11.8 휠 받기 (기본 PyPI + 추가 인덱스)
--extra-index-url https://download.pytorch.org/whl/cu118

########################
# 서버/웹 기본
########################
fastapi==0.104.1
uvicorn[standard]==0.24.0
python-multipart==0.0.6
aiofiles==23.2.1
asyncio-throttle==1.0.2
requests==2.31.0
psutil==5.9.6
websockets

########################
# 이미지/수치
########################
pillow==10.0.1
# TensorFlow 2.12.0은 numpy<1.24 요구 → 충돌 방지 위해 범위 고정
numpy==1.26.4
opencv-python==4.8.1.78

########################
# 설정/검증
########################
pydantic==2.7.4
pydantic-settings==2.8.1

########################
# DL 프레임워크
########################
# TensorFlow (GPU 포함 단일 패키지)
# tensorflow==2.12.0

# PyTorch (CUDA 11.8 휠)
torch==2.2.2
torchvision==0.17.2

########################
# 인페인팅/배경제거
########################
rembg==2.0.50
simple-lama-inpainting

# rembg 가속용 (CUDA 11.8 빌드)
onnxruntime-gpu==1.16.3

########################
# GPU 모니터링
########################
pynvml==11.5.0
# nvidia-ml-py3==7.352.0  # ← 기능 중복이라 제거 권장

########################
# (선택) CUDA 커널 직접 호출 시만
########################
#pycuda==2022.2.2