scorevision: push artifact

chute_config.yml

@@ -2,30 +2,11 @@ Image:
   from_base: parachutes/python:3.12
   run_command:
     - pip install --upgrade setuptools wheel
-    - >-
-      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
+    - 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'
 
 NodeSelector:
   gpu_count: 1
   min_vram_gb_per_gpu: 16
-  include:
-    - 4090
-    - a100
-    - a100_sxm
-    - h100
-    - h100_sxm
-    - l40s
-    - a40
 
 Chute:
   timeout_seconds: 300