[build] Add CUDA 12/13 build matrix with full GPU compatibility

Breakthrough · Breakthrough · commit c3c161b48f79 · 2026-01-27T18:31:11.000-05:00
diff --git a/.github/workflows/build_wheels_windows.yml b/.github/workflows/build_wheels_windows.yml
@@ -33,19 +33,45 @@ on:
 
 jobs:
   Build:
-    runs-on: windows-2025
+    runs-on: ${{ matrix.runs-on }}
     strategy:
       fail-fast: false
       matrix:
         python-version: ['3.13']
         platform: [x64]
+        cuda-version: ['12', '13']
+        include:
+          # CUDA 12: Supports Maxwell (5.0) through Blackwell (10.0)
+          # Uses windows-2025 runner
+          - cuda-version: '12'
+            runs-on: 'windows-2025'
+            cuda-installer: 'cuda_12.9.1_windows_network.exe'
+            cuda-path-version: 'v12.9'
+            cudnn-archive: 'cudnn-windows-x86_64-9.18.1.3_cuda12-archive.zip'
+            cudnn-folder: 'cudnn-windows-x86_64-9.18.1.3_cuda12-archive'
+            cuda-arch-bin: '5.0;5.2;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0;10.0'
+            cuda-arch-ptx: '10.0'
+            cache-key: 'nvidia-deps-cuda-12.9.1-cudnn-9.18.1.3'
+          # CUDA 13: Supports Turing (7.5) through Blackwell (12.0)
+          # Uses windows-2022 runner for MSVC 193x compatibility
+          - cuda-version: '13'
+            runs-on: 'windows-2022'
+            cuda-installer: 'cuda_13.1.1_windows_network.exe'
+            cuda-path-version: 'v13.1'
+            cudnn-archive: 'cudnn-windows-x86_64-9.18.1.3_cuda13-archive.zip'
+            cudnn-folder: 'cudnn-windows-x86_64-9.18.1.3_cuda13-archive'
+            cuda-arch-bin: '7.5;8.0;8.6;8.9;9.0;10.0;12.0'
+            cuda-arch-ptx: '12.0'
+            cache-key: 'nvidia-deps-cuda-13.1.1-cudnn-9.18.1.3'
     env:
       ACTIONS_ALLOW_UNSECURE_COMMANDS: true
       SDIST: 0
       ENABLE_HEADLESS: 0
       ENABLE_CONTRIB: 1
       ENABLE_ROLLING: ${{ inputs.rolling_build && 1 || 0 }}
       OPENCV_TEST_DATA_PATH: ${{ github.workspace }}\opencv_extra\testdata
+      CUDA_ARCH_BIN: ${{ matrix.cuda-arch-bin }}
+      CUDA_ARCH_PTX: ${{ matrix.cuda-arch-ptx }}
     steps:
     - name: Cleanup
       shell: bash
@@ -72,7 +98,7 @@ jobs:
       uses: actions/cache@v3
       with:
         path: .deps/Nvidia
-        key: nvidia-deps-cuda-12.4.0-cudnn-8.9.7.29
+        key: ${{ matrix.cache-key }}
     - name: Clone NVIDIA dependencies
       if: steps.cache-nvidia-deps.outputs.cache-hit != 'true'
       env:
@@ -85,27 +111,26 @@ jobs:
       shell: bash
     - name: 🔧 Install NVIDIA CUDA Toolkit
       run: |
-        $installer_path = ".deps/Nvidia/cuda_12.4.0_windows_network.exe"
+        $installer_path = ".deps/Nvidia/${{ matrix.cuda-installer }}"
         if (-not (Test-Path $installer_path)) {
           throw "CUDA Toolkit installer not found at $installer_path"
         }
-        echo "Installing CUDA Toolkit silently..."
-        $arguments = "-s nvcc_12.4 cudart_12.4 cublas_12.4 cublas_dev_12.4 cufft_12.4 cufft_dev_12.4 npp_12.4 npp_dev_12.4"
-        Start-Process -FilePath $installer_path -ArgumentList $arguments -Wait -NoNewWindow
+        echo "Installing CUDA ${{ matrix.cuda-version }} Toolkit silently..."
+        Start-Process -FilePath $installer_path -ArgumentList "-s" -Wait -NoNewWindow
         echo "Adding CUDA to PATH..."
-        $CUDA_PATH = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4"
+        $CUDA_PATH = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\${{ matrix.cuda-path-version }}"
         echo "CUDA_PATH=$CUDA_PATH" | Out-File -FilePath $env:GITHUB_ENV -Append
         Copy-Item -Path "$CUDA_PATH/bin/*" -Destination .  -Include "*.dll"
       shell: pwsh
     - name: 🔧 Install NVIDIA CuDNN
       run: |
-        $cudnn_path = ".deps/Nvidia/cudnn-windows-x86_64-8.9.7.29_cuda12-archive.zip"
+        $cudnn_path = ".deps/Nvidia/${{ matrix.cudnn-archive }}"
         if (-not (Test-Path $cudnn_path)) {
           throw "CuDNN archive not found at $cudnn_path"
         }
-        echo "Installing CuDNN..."
+        echo "Installing CuDNN for CUDA ${{ matrix.cuda-version }}..."
         7z x $cudnn_path
-        $CUDNN_PATH = "D:/a/opencv-python-cuda/opencv-python-cuda/cudnn-windows-x86_64-8.9.7.29_cuda12-archive"
+        $CUDNN_PATH = "D:/a/opencv-python-cuda/opencv-python-cuda/${{ matrix.cudnn-folder }}"
         echo "CUDNN_LIBRARY=$CUDNN_PATH/lib/x64/cudnn.lib" | Out-File -FilePath $env:GITHUB_ENV -Append
         echo "CUDNN_INCLUDE_DIR=$CUDNN_PATH/include" | Out-File -FilePath $env:GITHUB_ENV -Append
         Copy-Item -Path "$CUDNN_PATH/bin/*" -Destination .  -Include "*.dll"
@@ -116,9 +141,9 @@ jobs:
       if: ${{ inputs.restore_build_cache && !inputs.rolling_build }}
       with:
         path: _skbuild
-        key: ${{ runner.os }}-${{ matrix.python-version }}-${{ hashFiles('**/CMakeLists.txt') }}
+        key: ${{ runner.os }}-${{ matrix.python-version }}-cuda${{ matrix.cuda-version }}-${{ hashFiles('**/CMakeLists.txt') }}
         restore-keys: |
-          ${{ runner.os }}-${{ matrix.python-version }}-
+          ${{ runner.os }}-${{ matrix.python-version }}-cuda${{ matrix.cuda-version }}-
 
     - name: Build a package
       # CMake 3.25 regression fix. See https://stackoverflow.com/questions/74162633/problem-compiling-from-source-opencv-with-mvsc2019-in-64-bit-version
@@ -140,11 +165,11 @@ jobs:
       if: ${{ inputs.save_build_cache && !inputs.rolling_build }}
       with:
         path: _skbuild
-        key: ${{ runner.os }}-${{ matrix.python-version }}-${{ hashFiles('**/CMakeLists.txt') }}
+        key: ${{ runner.os }}-${{ matrix.python-version }}-cuda${{ matrix.cuda-version }}-${{ hashFiles('**/CMakeLists.txt') }}
     - name: Saving all wheels
       uses: actions/upload-artifact@v4
       with:
-        name: wheel-cuda-${{ matrix.platform }}
+        name: wheel-cuda${{ matrix.cuda-version }}-${{ matrix.platform }}
         path: wheelhouse/opencv*
 
   Test:
@@ -158,6 +183,7 @@ jobs:
       matrix:
         python-version: ['3.12']
         platform: [x64]
+        cuda-version: ['12', '13']
     env:
       ACTIONS_ALLOW_UNSECURE_COMMANDS: true
       OPENCV_TEST_DATA_PATH: ${{ github.workspace }}\opencv_extra\testdata
@@ -182,7 +208,7 @@ jobs:
     - name: Download a wheel accordingly to matrix
       uses: actions/download-artifact@v4
       with:
-        name: wheel-cuda-${{ matrix.platform }}
+        name: wheel-cuda${{ matrix.cuda-version }}-${{ matrix.platform }}
         path: wheelhouse/
     - name: Package installation
       run: |
diff --git a/.github/workflows/cache_nvidia_deps.yml b/.github/workflows/cache_nvidia_deps.yml
@@ -0,0 +1,45 @@
+# Simple workflow to cache NVIDIA dependencies
+# Run this once to populate the cache, then disable or delete
+
+name: Cache NVIDIA Dependencies
+
+on:
+  workflow_dispatch:
+
+jobs:
+  cache-deps:
+    runs-on: windows-2025
+    strategy:
+      matrix:
+        include:
+          - cuda-version: '12'
+            cache-key: 'nvidia-deps-cuda-12.9.1-cudnn-9.18.1.3'
+          - cuda-version: '13'
+            cache-key: 'nvidia-deps-cuda-13.1.1-cudnn-9.18.1.3'
+    steps:
+    - name: Checkout
+      uses: actions/checkout@v4
+      with:
+        submodules: false
+        fetch-depth: 1
+    - name: Cache NVIDIA dependencies (CUDA ${{ matrix.cuda-version }})
+      id: cache-nvidia-deps
+      uses: actions/cache@v3
+      with:
+        path: .deps/Nvidia
+        key: ${{ matrix.cache-key }}
+    - name: Clone NVIDIA dependencies
+      if: steps.cache-nvidia-deps.outputs.cache-hit != 'true'
+      env:
+        SSH_PRIVATE_KEY: ${{ secrets.DEPS_REPO_SSH_KEY }}
+      run: |
+        eval "$(ssh-agent -s)"
+        ssh-add - <<< "${SSH_PRIVATE_KEY}"
+        ssh-keyscan github.com >> ~/.ssh/known_hosts 2>/dev/null
+        git clone git@github.com:Breakthrough/opencv-python-cuda-deps.git .deps
+      shell: bash
+    - name: Cache populated
+      run: |
+        echo "NVIDIA dependencies for CUDA ${{ matrix.cuda-version }} cached successfully!"
+        ls -la .deps/Nvidia/ || echo "Cache was already populated"
+      shell: bash
diff --git a/setup.py b/setup.py
@@ -22,19 +22,14 @@ def main():
     build_java = "ON" if get_build_env_var_by_name("java") else "OFF"
     build_rolling = get_build_env_var_by_name("rolling")
 
-    # TODO(@Breakthrough): What architectures should we ship?
-    # fermi     = "2.0"
-    # kepler    = "3.0;3.5;3.7"
-    # maxwell   = "5.0;5.2"
-    # pascal    = "6.0;6.1"
-    # volta     = "7.0"
-    # turing    = "7.5"
-    # ampere    = "8.0;8.6"
-    # lovelace  = "8.9"
-    # hopper    = "9.0"
-    # blackwell =  "10.0;12.0"
-    cuda_arch_bin = "5.0;5.2;6.0;6.1;7.0;7.5,8.0"
-    cuda_arch_ptx = "8.0"
+    # CUDA architecture configuration - can be overridden via environment variables
+    # CUDA 12.x supports: 5.0+ (Maxwell through Hopper)
+    # CUDA 13.x supports: 7.5+ (Turing through Hopper)
+    # Reference architectures:
+    #   maxwell=5.0;5.2, pascal=6.0;6.1, volta=7.0, turing=7.5,
+    #   ampere=8.0;8.6, lovelace=8.9, hopper=9.0
+    cuda_arch_bin = os.environ.get("CUDA_ARCH_BIN", "7.5;8.0;8.6;8.9;9.0")
+    cuda_arch_ptx = os.environ.get("CUDA_ARCH_PTX", "9.0")
 
     # NOTE: since 2.3.0 numpy upgraded from manylinux2014 to manylinux_2_28
     # see https://numpy.org/doc/stable/release/2.3.0-notes.html#numpy-2-3-0-release-notes
