worlabel/ai/environment.yml
2024-09-05 17:07:25 +09:00

18 lines
277 B
YAML

name: worlabel_ai_env
channels:
- conda-forge
- nvidia
- pytorch
- defaults
dependencies:
- python=3.10.10
- pytorch=2.3.1
- torchvision=0.16.1
- torchaudio=2.3.1
- pytorch-cuda=12.1
- fastapi
- uvicorn
- ultralytics
- dill
- boto3
- python-dotenv