[dist] Build module with CUDA enabled

Author: Breakthrough

.github/workflows/build_wheels_windows.yml

@@ -1,4 +1,13 @@
-name: Windows x86_64
+# GitHub Actions Workflow: Build opencv-python with CUDA support on Windows
+#
+# This workflow compiles opencv-python from source with CUDA enabled on a
+# GitHub-hosted Windows runner. The resulting Python wheel is then uploaded
+# as a build artifact.
+#
+# This is a complex and long-running process. It is configured to run only
+# on manual trigger (workflow_dispatch).
+
+name: Windows CUDA x64
 
 on:
   workflow_dispatch:
@@ -48,6 +57,58 @@ jobs:
       uses: microsoft/setup-msbuild@v1.1
     - name: Setup NASM
       uses: ilammy/setup-nasm@v1
+    - name: Cache CUDA Toolkit Installer
+      id: cache-cuda-installer
+      uses: actions/cache@v3
+      with:
+        path: ./.cache/cuda_installer.exe
+        key: cuda-installer-12.0.0
+    - name: 🔧 Install NVIDIA CUDA Toolkit
+      run: |
+        $installer_path = "./.cache/cuda_installer.exe"
+        if (-not (Test-Path $installer_path)) {
+          echo "Downloading CUDA Toolkit..."
+          $cuda_installer_url = "https://developer.download.nvidia.com/compute/cuda/12.0.0/network_installers/cuda_12.0.0_windows_network.exe"
+          New-Item -Path (Split-Path $installer_path) -ItemType Directory -Force
+          curl -L -o $installer_path $cuda_installer_url
+        } else {
+          echo "CUDA Toolkit installer found in cache."
+        }
+        echo "Installing CUDA Toolkit silently..."
+        $arguments = "-s nvcc_12.0 cudart_12.0"
+        Start-Process -FilePath $installer_path -ArgumentList $arguments -Wait -NoNewWindow
+        echo "Adding CUDA to PATH..."
+        $CUDA_PATH = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.0"
+        echo "CUDA_PATH=$CUDA_PATH" | Out-File -FilePath $env:GITHUB_ENV -Append
+        echo "$CUDA_PATH\bin" | Out-File -FilePath $env:GITHUB_PATH -Append
+        echo "$CUDA_PATH\lib\x64" | Out-File -FilePath $env:GITHUB_PATH -Append
+        echo "$CUDA_PATH"
+      shell: pwsh
+    - name: Cache CuDNN Installer
+      id: cache-cudnn-installer
+      uses: actions/cache@v3
+      with:
+        path: ./.cache/cudnn_installer.exe
+        key: cudnn-installer-8.8.0.121-windows
+    - name: 🔧 Install NVIDIA CuDNN
+      run: |
+        $installer_path = "./.cache/cudnn_installer.exe"
+        if (-not (Test-Path $installer_path)) {
+          echo "Downloading CuDNN..."
+          $cudnn_installer_url = "https://developer.download.nvidia.com/compute/redist/cudnn/v8.8.0/local_installers/12.0/cudnn_8.8.0.121_windows.exe"
+          New-Item -Path (Split-Path $installer_path) -ItemType Directory -Force
+          curl -L -o $installer_path $cudnn_installer_url
+        } else {
+          echo "CuDNN installer found in cache."
+        }
+        echo "Installing CuDNN silently..."
+        # The CuDNN local installer for Windows is a self-extracting executable.
+        # We assume it installs to the correct CUDA Toolkit directory.
+        # We use -s for silent installation.
+        Start-Process -FilePath $installer_path -ArgumentList "-s" -Wait -NoNewWindow
+        # TODO: Set CUDNN_LIBRARY CUDNN_INCLUDE_DIR so the build works correctly.
+      shell: pwsh
+
     - 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
       run: |

setup.py

@@ -86,8 +86,8 @@ def main():
     if build_headless and not build_contrib:
         package_name = "opencv-python-headless"
 
-    if build_rolling:
-        package_name += "-rolling"
+    #if build_rolling:
+    #    package_name += "-rolling"
 
     package_name = os.environ.get('OPENCV_PYTHON_PACKAGE_NAME', package_name)
 
@@ -189,6 +189,10 @@ def main():
             "-DBUILD_DOCS=OFF",
             "-DPYTHON3_LIMITED_API=ON",
             "-DBUILD_OPENEXR=ON",
+            "-DWITH_CUDA=ON",
+            "-DCUDA_ARCH_BIN=\"6.0;6.1;7.0;7.5\"",
+            "-DCUDA_ARCH_PTX=\"7.5\"",
+            "-DOPENCV_ENABLE_NONFREE=ON",
         ]
         + (
             # CMake flags for windows/arm64 build