File size: 589 Bytes
05239d1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
Image:
from_base: parachutes/python:3.12
run_command:
- pip install --upgrade setuptools wheel
- pip install "torch==2.7.1" "torchvision==0.22.1"
- pip install "ultralytics==8.3.222" "opencv-python-headless" "numpy" "pydantic"
- pip install scikit-learn
- pip install onnxruntime-gpu
set_workdir: /app
NodeSelector:
gpu_count: 1
min_vram_gb_per_gpu: 24
exclude:
- "5090"
- b200
- h200
- mi300x
Chute:
timeout_seconds: 900
concurrency: 4
max_instances: 5
scaling_threshold: 0.5
shutdown_after_seconds: 3600 |