Merge branch 'ai/chore/env-setup' into 'ai/develop'

Chore: conda로 의존성 관리 추가

See merge request s11-s-project/S11P21S002!45
This commit is contained in:
김태수 2024-09-05 09:49:23 +09:00
commit 1f63944ac1
3 changed files with 38 additions and 11 deletions

View File

@ -1,15 +1,22 @@
# FastAPI를 이용한 AI 모델 관련 API
## app
## conda 환경 세팅
```bash
conda env create -f environment.yml
conda activate worlabel_ai_env
```
### api
- fastapi 호출 함수 정의
## FastAPI Project 구조
### schemas
### app/api
- api 호출 라우터 정의
### app/schemas
- api의 request/response 등 Pydantic 모델 정의
### services
- 필요한 ai 서비스들을 정의
### app/services
- AI 관련 패키지를 이용하는 메서드 정의
### app/utils
- 프로젝트 전역에서 이용하는 formatter 등 정의
### utils
- 프로젝트 전역에서 이용하는 formatter 등 정의

15
ai/environment.yml Normal file
View File

@ -0,0 +1,15 @@
name: worlabel_ai_env
channels:
- conda-forge
- nvidia
- pytorch
- defaults
dependencies:
- python=3.10.10
- pytorch
- torchvision
- torchaudio
- pytorch-cuda=12.1
- fastapi
- uvicorn
- ultralytics

View File

@ -1,7 +1,12 @@
fastapi==0.104.1
uvicorn==0.30.6
torch==2.4.0 -f https://download.pytorch.org/whl/cpu
torchaudio==2.4.0 -f https://download.pytorch.org/whl/cpu
torchvision==0.19.0 -f https://download.pytorch.org/whl/cpu
ultralytics==8.2.82
ultralytics-thop==2.0.5
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.4.0+cu121
--extra-index-url https://download.pytorch.org/whl/cu121
torchaudio==2.4.0+cu121
--extra-index-url https://download.pytorch.org/whl/cu121
torchvision==0.19.0+cu121