Chore: conda로 의존성 관리 추가
- READ.me 및 노션 문서에 세팅 방법 적어둠 - conda로 관리하는 것이 Cuda 설치 등 더 유연함 - gpu 서버 또한 conda로 가상 환경 설정 예정
This commit is contained in:
parent
635c2a2033
commit
27e60ee68a
21
ai/README.md
21
ai/README.md
@ -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
15
ai/environment.yml
Normal 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
|
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user