63 lines
1.4 KiB
Plaintext
63 lines
1.4 KiB
Plaintext
# ============================
|
|
# 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
|