meaculpitt commited on
Commit
f7cb518
·
verified ·
1 Parent(s): 786da4f

scorevision: push artifact

Browse files
Files changed (1) hide show
  1. config.yml +9 -1
config.yml CHANGED
@@ -2,11 +2,19 @@ Image:
2
  from_base: parachutes/python:3.12
3
  run_command:
4
  - pip install --upgrade setuptools wheel
5
- - pip install 'numpy>=1.23' 'onnxruntime>=1.16' 'opencv-python>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9' 'torch>=2.8'
6
 
7
  NodeSelector:
8
  gpu_count: 1
9
  min_vram_gb_per_gpu: 16
 
 
 
 
 
 
 
 
10
 
11
  Chute:
12
  timeout_seconds: 300
 
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' 'nvidia-cudnn-cu12' 'nvidia-cublas-cu12' 'opencv-python-headless>=4.7' 'pillow>=9.5' 'huggingface_hub>=0.19.4' 'pydantic>=2.0' 'pyyaml>=6.0' 'aiohttp>=3.9'
6
 
7
  NodeSelector:
8
  gpu_count: 1
9
  min_vram_gb_per_gpu: 16
10
+ include:
11
+ - "4090"
12
+ - a100
13
+ - a100_sxm
14
+ - h100
15
+ - h100_sxm
16
+ - l40s
17
+ - a40
18
 
19
  Chute:
20
  timeout_seconds: 300