scorevision: push artifact

chute_config.yml

@@ -2,11 +2,30 @@ Image:
   from_base: parachutes/python:3.12
   run_command:
     - pip install --upgrade setuptools wheel
-    - 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'
+    - >-
+      pip install 'numpy>=1.23' 'onnxruntime-gpu>=1.16'
+      'opencv-python-headless>=4.7' 'pillow>=9.5'
+      'huggingface_hub>=0.19.4' 'pydantic>=2.0'
+      'pyyaml>=6.0' 'aiohttp>=3.9'
+    - >-
+      python3 -c "
+      import os, nvidia.cudnn, nvidia.cublas;
+      cudnn=os.path.join(os.path.dirname(nvidia.cudnn.__file__),'lib');
+      cublas=os.path.join(os.path.dirname(nvidia.cublas.__file__),'lib');
+      open('/etc/ld.so.conf.d/nvidia-ort.conf','w').write(cudnn+chr(10)+cublas+chr(10))
+      " && ldconfig
 
 NodeSelector:
   gpu_count: 1
   min_vram_gb_per_gpu: 16
+  include:
+    - 4090
+    - a100
+    - a100_sxm
+    - h100
+    - h100_sxm
+    - l40s
+    - a40
 
 Chute:
   timeout_seconds: 300