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

15 lines
221 B
YAML

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