worlabel/ai/requirements.txt
김진현 27e60ee68a Chore: conda로 의존성 관리 추가
- READ.me 및 노션 문서에 세팅 방법 적어둠
- conda로 관리하는 것이 Cuda 설치 등 더 유연함
- gpu 서버 또한 conda로 가상 환경 설정 예정
2024-09-04 22:26:22 +09:00

13 lines
319 B
Plaintext

fastapi==0.104.1
uvicorn==0.30.6
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