mars44 commited on
Commit
1758982
·
verified ·
1 Parent(s): b7a3d62

scorevision: push artifact

Browse files
Files changed (1) hide show
  1. chute_config.yml +7 -2
chute_config.yml CHANGED
@@ -2,12 +2,17 @@ Image:
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
- - pip install 'numpy>=1.23' 'onnxruntime-gpu>=1.16' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
6
- - pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu128
 
 
 
 
7
 
8
  NodeSelector:
9
  gpu_count: 1
10
  min_vram_gb_per_gpu: 16
 
11
  include:
12
  - pro_6000
13
 
 
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
+ # CPU onnxruntime pinned to the compliance checker's version: same kernels, so the live chute
6
+ # and the offline audit re-run produce identical boxes.
7
+ - pip install 'numpy>=1.23,<3' 'onnxruntime==1.26.0' 'opencv-python-headless>=4.8' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9' 'huggingface_hub>=0.24'
8
+ # torch is imported by the chute wrapper, not by miner.py
9
+ - pip install torch==2.8.0 --index-url https://download.pytorch.org/whl/cu128
10
+ set_workdir: /app
11
 
12
  NodeSelector:
13
  gpu_count: 1
14
  min_vram_gb_per_gpu: 16
15
+ max_hourly_price_per_gpu: 2
16
  include:
17
  - pro_6000
18