scorevision: push artifact

miner.py

@@ -45,11 +45,27 @@ class Miner:
         if not self.onnx_path.exists():
             raise FileNotFoundError(f'Model not found at {self.onnx_path}')
 
+        # Help ORT find CUDA libs shipped with nvidia-*-cu12 packages (pytorch/onnxruntime).
+        import os as _os
+        import site as _site
+        import glob as _glob
+        cuda_lib_dirs: list[str] = []
+        for sp in _site.getsitepackages() + [_site.getusersitepackages()]:
+            for sub in ('nvidia/cuda_runtime/lib', 'nvidia/cublas/lib', 'nvidia/cudnn/lib',
+                        'nvidia/cufft/lib', 'nvidia/cuda_nvrtc/lib', 'nvidia/curand/lib',
+                        'nvidia/cusparse/lib', 'nvidia/cusolver/lib', 'nvidia/nvjitlink/lib'):
+                p = f'{sp}/{sub}'
+                if _glob.glob(f'{p}/*.so*'):
+                    cuda_lib_dirs.append(p)
+        if cuda_lib_dirs:
+            existing = _os.environ.get('LD_LIBRARY_PATH', '')
+            _os.environ['LD_LIBRARY_PATH'] = ':'.join(cuda_lib_dirs + ([existing] if existing else []))
+
         providers: list = []
         try:
             ort.preload_dlls()
-        except Exception:
-            pass
+        except Exception as _pe:
+            print(f'[Miner] preload_dlls failed: {_pe}')
         available = ort.get_available_providers()
         if 'CUDAExecutionProvider' in available:
             providers.append(('CUDAExecutionProvider', {'device_id': 0}))
@@ -61,10 +77,13 @@ class Miner:
         inp = self.session.get_inputs()[0]
         self.input_shape = inp.shape
         self.input_dtype = np.float16 if inp.type == 'tensor(float16)' else np.float32
-        print(f'[Miner] Loaded {self.onnx_path.name} | providers={self.session.get_providers()} | dtype={self.input_dtype}')
+        self.active_providers = self.session.get_providers()
+        print(f'[Miner] Loaded {self.onnx_path.name} | providers={self.active_providers} | dtype={self.input_dtype}')
+        print(f'[Miner] cuda_lib_dirs discovered: {cuda_lib_dirs[:3]}')
+        print(f'[Miner] ort.get_available_providers() = {available}')
 
     def __repr__(self) -> str:
-        return f'PetrolMiner(yolo11s-onnx-fp16-nms, tta={self.USE_TTA}, conf={self.CONF_THRES})'
+        return f'PetrolMiner(yolo11s-onnx-fp16-nms, tta={self.USE_TTA}, conf={self.CONF_THRES}, providers={getattr(self, "active_providers", "?")})'
 
     @staticmethod
     def _letterbox(img, new_size=1280, color=(114, 114, 114)):