opensportslib 0.3.0.dev16__tar.gz → 0.3.0.dev17__tar.gz

{opensportslib-0.3.0.dev16/opensportslib.egg-info → opensportslib-0.3.0.dev17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.3.0.dev16
+Version: 0.3.0.dev17
 Summary: OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data.
 Author: Jeet Vora
 Requires-Python: >=3.12

{opensportslib-0.3.0.dev16 → opensportslib-0.3.0.dev17}/opensportslib/setup/setup.py

@@ -1,3 +1,4 @@
+import platform
 import subprocess
 import sys
 
@@ -7,9 +8,16 @@ CUDA_WHEEL_VERSIONS = {
     "cu128": (12, 8),
     "cu130": (13, 0),
 }
-MIN_SUPPORTED_COMPUTE_CAPABILITY = (7, 0)
+MIN_SUPPORTED_COMPUTE_CAPABILITY = (5, 0)
 LEGACY_GPU_MAX_COMPUTE_CAPABILITY = (7, 4)
 LEGACY_GPU_CUDA_WHEEL = "cu126"
+LEGACY_GPU_CUDA_WHEEL_MAX_COMPUTE_CAPABILITY = (9, 0)
+CUDA13_REQUIRED_MIN_COMPUTE_CAPABILITY = (10, 0)
+LEGACY_GPU_TORCH_PACKAGES = (
+    "torch==2.10.0",
+    "torchvision==0.25.0",
+    "torchaudio==2.10.0",
+)
 
 XVARS_DEPENDENCY_PINS = {
     "transformers": "4.38.2",
@@ -83,12 +91,41 @@ def select_cuda_wheel(cuda_version, compute_capabilities):
     compatible_tags = [
         tag for tag, wheel_version in CUDA_WHEEL_VERSIONS.items() if wheel_version <= driver_version
     ]
-    if compute_capabilities and any(
+    has_legacy_gpu = compute_capabilities and any(
         capability <= LEGACY_GPU_MAX_COMPUTE_CAPABILITY for capability in compute_capabilities
-    ):
+    )
+    if has_legacy_gpu:
+        if platform.machine().lower() in {"aarch64", "arm64"}:
+            raise RuntimeError(
+                "Official Linux ARM64 CUDA 12.6 PyTorch wheels support Ampere and newer GPUs only. "
+                "Use an x86_64 host or build PyTorch from source for the visible legacy GPU architecture."
+            )
+        if any(
+            capability > LEGACY_GPU_CUDA_WHEEL_MAX_COMPUTE_CAPABILITY
+            for capability in compute_capabilities
+        ):
+            raise RuntimeError(
+                "Visible GPUs require incompatible prebuilt PyTorch wheels. Set CUDA_VISIBLE_DEVICES to "
+                "either the legacy GPUs or the newer GPUs, then run setup again."
+            )
         compatible_tags = [tag for tag in compatible_tags if tag == LEGACY_GPU_CUDA_WHEEL]
 
+    if any(
+        capability >= CUDA13_REQUIRED_MIN_COMPUTE_CAPABILITY
+        for capability in compute_capabilities
+    ):
+        compatible_tags = [tag for tag in compatible_tags if tag == "cu130"]
+
     if not compatible_tags:
+        if any(
+            capability >= CUDA13_REQUIRED_MIN_COMPUTE_CAPABILITY
+            for capability in compute_capabilities
+        ):
+            raise RuntimeError(
+                "The visible GPU architecture requires the CUDA 13.0 PyTorch wheel. "
+                "Update the NVIDIA driver so nvidia-smi reports CUDA 13.0 or newer, "
+                "then run setup again."
+            )
         raise RuntimeError(
             f"CUDA {cuda_version} is too old for the supported PyTorch wheels: "
             f"{', '.join(CUDA_WHEEL_VERSIONS)}."
@@ -102,6 +139,14 @@ def select_cuda_wheel(cuda_version, compute_capabilities):
     return selected
 
 
+def select_torch_packages(compute_capabilities):
+    if compute_capabilities and any(
+        capability <= LEGACY_GPU_MAX_COMPUTE_CAPABILITY for capability in compute_capabilities
+    ):
+        return LEGACY_GPU_TORCH_PACKAGES
+    return ("torch", "torchvision", "torchaudio")
+
+
 CUDA_VERSION, _DETECTED_CUDA_TAG = get_cuda_version()
 GPU_COMPUTE_CAPABILITIES = get_gpu_compute_capabilities()
 CUDA_TAG = select_cuda_wheel(CUDA_VERSION, GPU_COMPUTE_CAPABILITIES)
@@ -121,14 +166,15 @@ def install_xvars_dependencies(DEPENDENCY_PINS):
 def install_torch():
     python = sys.executable
     subprocess.call([python, "-m", "pip", "uninstall", "-y", "torch", "torchvision", "torchaudio"])
+    packages = select_torch_packages(GPU_COMPUTE_CAPABILITIES)
 
     subprocess.check_call([
         python, "-m", "pip", "install",
-        "torch", "torchvision", "torchaudio",
+        *packages,
         "--index-url",
         f"https://download.pytorch.org/whl/{CUDA_TAG}",
     ])
-    print(f"\nSuccess with {CUDA_TAG}")
+    print(f"\nSuccess with {CUDA_TAG}: {', '.join(packages)}")
     return CUDA_TAG
 
 def install_dali():

{opensportslib-0.3.0.dev16 → opensportslib-0.3.0.dev17/opensportslib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.3.0.dev16
+Version: 0.3.0.dev17
 Summary: OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data.
 Author: Jeet Vora
 Requires-Python: >=3.12

{opensportslib-0.3.0.dev16 → opensportslib-0.3.0.dev17}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opensportslib"
-version = "0.3.0.dev16"
+version = "0.3.0.dev17"
 description = "OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data."
 readme = "README.md"
 requires-python = ">=3.12"

{opensportslib-0.3.0.dev16 → opensportslib-0.3.0.dev17}/tests/test_setup_cli.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+import pytest
+
 from opensportslib import cli
 from opensportslib.setup import setup as setup_lib
 
@@ -8,10 +10,27 @@ def test_select_cuda_wheel_uses_cu126_for_pre_sm75_gpu_with_cuda_13():
     assert setup_lib.select_cuda_wheel("13.0", [(7, 0)]) == "cu126"
 
 
+def test_select_cuda_wheel_uses_cu126_for_pascal_with_cuda_13():
+    assert setup_lib.select_cuda_wheel("13.0", [(6, 0)]) == "cu126"
+
+
+def test_select_torch_packages_pins_pre_sm75_gpu_compatibility_stack():
+    assert setup_lib.select_torch_packages([(7, 0)]) == (
+        "torch==2.10.0",
+        "torchvision==0.25.0",
+        "torchaudio==2.10.0",
+    )
+
+
 def test_select_cuda_wheel_uses_cu130_for_dgx_spark():
     assert setup_lib.select_cuda_wheel("13.0", [(12, 1)]) == "cu130"
 
 
+def test_select_cuda_wheel_rejects_new_architecture_without_cuda_13_driver():
+    with pytest.raises(RuntimeError, match="CUDA 13.0"):
+        setup_lib.select_cuda_wheel("12.8", [(10, 0)])
+
+
 def test_select_cuda_wheel_uses_highest_driver_compatible_wheel():
     assert setup_lib.select_cuda_wheel("12.8", [(8, 0)]) == "cu128"