Chore: conda로 의존성 관리 추가

- READ.me 및 노션 문서에 세팅 방법 적어둠
- conda로 관리하는 것이 Cuda 설치 등 더 유연함
- gpu 서버 또한 conda로 가상 환경 설정 예정
This commit is contained in:
김진현 2024-09-04 22:26:22 +09:00
parent 635c2a2033
commit 27e60ee68a
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 관련 패키지를 이용하는 메서드 정의
### utils
### app/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