PyNvVideoCodec 2.2.1__cp312-cp312-win_arm64.whl

samples/requirements.txt

@@ -0,0 +1,84 @@
+# This copyright notice applies to this file only
+#
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: MIT
+#
+# Permission is hereby granted, free of charge, to any person obtaining a
+# copy of this software and associated documentation files (the "Software"),
+# to deal in the Software without restriction, including without limitation
+# the rights to use, copy, modify, merge, publish, distribute, sublicense,
+# and/or sell copies of the Software, and to permit persons to whom the
+# Software is furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
+# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+# DEALINGS IN THE SOFTWARE.
+
+# Requirements for PyNvVideoCodec Sample Applications
+# This file covers dependencies for:
+# - samples/basic/
+# - samples/advanced/
+# - samples/jupyter/
+
+# IMPORTANT: PyTorch must be installed with CUDA support.
+# Windows ARM64 uses NVIDIA's ARM CUDA Torch/TorchVision wheels.
+
+# Core dependencies (required for most samples)
+numpy>=1.21.0
+# Native Windows ARM64 users should run setup_pycuda_win_arm64.bat separately.
+pycuda>=2021.1; sys_platform != "win32" or platform_machine != "ARM64"
+
+# PyNvVideoCodec (assuming it's installed separately or available locally)
+# PyNvVideoCodec
+
+# For visualization in jupyter notebooks
+matplotlib>=3.3.0
+
+# For downloading sample videos in jupyter notebooks
+requests>=2.25.0
+
+# For running Jupyter notebooks
+jupyter>=1.0.0
+jupyterlab>=3.0.0
+ipython>=7.20.0
+
+# PyTorch with CUDA 13.0 support
+# See: https://download.pytorch.org/whl/cu130/torch/
+--extra-index-url https://download.pytorch.org/whl/cu130
+# Windows ARM64 Torch/TorchVision CUDA wheels are available for CPython 3.11-3.14.
+# See: https://pypi.nvidia.com/nvtorch_oot_nightly/
+--extra-index-url https://pypi.nvidia.com/nvtorch_oot_nightly
+torch==2.9.1; sys_platform != "win32" or platform_machine != "ARM64"
+torch==2.15.0.dev20260820+cu134; sys_platform == "win32" and platform_machine == "ARM64" and python_version >= "3.11" and python_version < "3.15"
+torchvision==0.30.0.dev20260820+cu134; sys_platform == "win32" and platform_machine == "ARM64" and python_version >= "3.11" and python_version < "3.15"
+
+
+# Note: Standard library modules used (no installation needed):
+# - json, os, sys, pathlib, argparse, time, threading, multiprocessing
+# - queue, ctypes, pickle, datetime, gc, atexit
+
+
+# Running Jupyter Notebooks:
+#
+# Option 1 - Using Jupyter Notebook (Classic):
+#    jupyter notebook
+#    # Then navigate to samples/jupyter/ and open the desired .ipynb file
+#
+# Option 2 - Using JupyterLab (Recommended):
+#    jupyter lab
+#    # Then navigate to samples/jupyter/ and open the desired .ipynb file
+#
+# Option 3 - Using specific notebook:
+#    jupyter notebook samples/jupyter/object_detection_tutorial.ipynb
+#    jupyter notebook samples/jupyter/simple_decode_tutorial.ipynb
+#    jupyter notebook samples/jupyter/face_protect_tutorial.ipynb
+#
+# Note: Ensure PyNvVideoCodec is installed before running the notebooks
+

samples/setup_pycuda_win_arm64.bat

@@ -0,0 +1,531 @@
+@echo off
+rem This copyright notice applies to this file only
+
+rem SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+rem SPDX-License-Identifier: MIT
+
+rem Permission is hereby granted, free of charge, to any person obtaining a
+rem copy of this software and associated documentation files (the "Software"),
+rem to deal in the Software without restriction, including without limitation
+rem the rights to use, copy, modify, merge, publish, distribute, sublicense,
+rem and/or sell copies of the Software, and to permit persons to whom the
+rem Software is furnished to do so, subject to the following conditions:
+
+rem The above copyright notice and this permission notice shall be included in
+rem all copies or substantial portions of the Software.
+
+rem THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+rem IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+rem FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
+rem THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+rem LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+rem FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+rem DEALINGS IN THE SOFTWARE.
+
+setlocal EnableExtensions EnableDelayedExpansion
+
+rem Wheels or source checkouts can preserve this file with LF-only line endings.
+rem cmd.exe may then fail to resolve later "call :label" subroutines. Relaunch
+rem once from a CRLF-normalized temp copy before using any subroutine calls.
+if not defined PYNVC_PYCUDA_CRLF_NORMALIZED (
+    set "PYNVC_PYCUDA_CRLF_NORMALIZED=1"
+    set "PYNVC_PYCUDA_ORIGINAL_BAT=%~f0"
+    set "PYNVC_PYCUDA_CRLF_BAT=%TEMP%\%~n0_%RANDOM%_%RANDOM%.bat"
+    powershell.exe -NoProfile -ExecutionPolicy Bypass -Command "$src=$env:PYNVC_PYCUDA_ORIGINAL_BAT; $dst=$env:PYNVC_PYCUDA_CRLF_BAT; $text=[System.IO.File]::ReadAllText($src); $text=[System.Text.RegularExpressions.Regex]::Replace($text, '\r?\n', [Environment]::NewLine); [System.IO.File]::WriteAllText($dst, $text, [System.Text.Encoding]::ASCII)"
+    if errorlevel 1 (
+        echo ERROR: Failed to create a CRLF-normalized copy of "%~f0".
+        exit /b 1
+    )
+    call "!PYNVC_PYCUDA_CRLF_BAT!" %*
+    set "PYNVC_PYCUDA_CRLF_STATUS=!ERRORLEVEL!"
+    del /q "!PYNVC_PYCUDA_CRLF_BAT!" >nul 2>nul
+    exit /b !PYNVC_PYCUDA_CRLF_STATUS!
+)
+set "PYCUDA_VERSION=2026.1"
+set "PYCUDA_TAG=v2026.1"
+set "PYCUDA_COMMIT=7610751702a013053fab8b604bdf1c236dd1f3c0"
+set "PYCUDA_REPO=https://github.com/inducer/pycuda.git"
+set "BPL_SUBSET_COMMIT=021f0d90d2730cc138b4dd476cb3184b27c2a33e"
+set "COMPYTE_COMMIT=f1010ef64b5d4559ca4bb7690d4b3394091477b6"
+set "PYTOOLS_VERSION=2024.1.14"
+
+if not defined PYTHON_EXE set "PYTHON_EXE=python"
+if not defined PYCUDA_BUILD_ROOT set "PYCUDA_BUILD_ROOT=%LOCALAPPDATA%\PyNvVideoCodec\pycuda-win-arm64\%PYCUDA_VERSION%"
+if not defined PYCUDA_SOURCE_DIR set "PYCUDA_SOURCE_DIR=%PYCUDA_BUILD_ROOT%\src"
+
+echo.
+echo PyNvVideoCodec PyCUDA setup for native Windows ARM64
+echo =====================================================
+echo.
+
+call :check_python
+if errorlevel 1 goto :fail
+
+call :activate_msvc
+if errorlevel 1 goto :fail
+
+call :find_cuda
+if errorlevel 1 goto :fail
+
+call :install_python_build_deps
+if errorlevel 1 goto :fail
+
+call :sync_pycuda_source
+if errorlevel 1 goto :fail
+
+call :write_siteconf
+if errorlevel 1 goto :fail
+
+call :install_pycuda
+if errorlevel 1 goto :fail
+
+call :install_cuda_dll_path_hook
+if errorlevel 1 goto :fail
+
+call :verify_pycuda
+if errorlevel 1 goto :fail
+
+echo.
+echo PyCUDA setup completed successfully.
+exit /b 0
+
+:check_python
+echo Checking Python...
+"%PYTHON_EXE%" -c "import platform, sys; print('  executable:', sys.executable); print('  version   :', sys.version.split()[0]); print('  machine   :', platform.machine()); raise SystemExit(0 if sys.implementation.name == 'cpython' and sys.maxsize > 2**32 and platform.machine().lower() in ('arm64', 'aarch64') else 1)"
+if errorlevel 1 (
+    echo ERROR: PYTHON_EXE must point to a native 64-bit CPython ARM64 interpreter.
+    echo        Set PYTHON_EXE to the ARM64 python.exe in your target environment and rerun this script.
+    exit /b 1
+)
+exit /b 0
+
+:activate_msvc
+if /I "%VSCMD_ARG_TGT_ARCH%"=="arm64" (
+    where cl.exe >nul 2>nul
+    if not errorlevel 1 (
+        where link.exe >nul 2>nul
+        if not errorlevel 1 (
+            echo Using active MSVC environment for target ARM64.
+            goto :msvc_verify
+        )
+    )
+    echo Existing MSVC target is ARM64, but cl.exe/link.exe are not on PATH.
+    echo Re-activating MSVC from vcvarsall.bat...
+)
+
+call :find_vcvarsall
+if errorlevel 1 exit /b 1
+
+echo Activating MSVC ARM64 build environment...
+call "%VCVARSALL%" arm64
+if errorlevel 1 (
+    echo ERROR: vcvarsall.bat failed for target arm64.
+    exit /b 1
+)
+
+:msvc_verify
+if /I not "%VSCMD_ARG_TGT_ARCH%"=="arm64" (
+    echo ERROR: MSVC target architecture is "%VSCMD_ARG_TGT_ARCH%", expected "arm64".
+    exit /b 1
+)
+
+where cl.exe >nul 2>nul
+if errorlevel 1 (
+    echo ERROR: cl.exe was not found after activating Visual Studio.
+    echo        Make sure this Visual Studio installation has the MSVC ARM64 C++ build tools installed.
+    exit /b 1
+)
+
+where link.exe >nul 2>nul
+if errorlevel 1 (
+    echo ERROR: link.exe was not found after activating Visual Studio.
+    echo        Make sure this Visual Studio installation has the MSVC ARM64 C++ build tools installed.
+    exit /b 1
+)
+
+echo MSVC target: %VSCMD_ARG_TGT_ARCH%
+exit /b 0
+
+:find_vcvarsall
+if defined VCVARSALL (
+    if exist "%VCVARSALL%" exit /b 0
+
+    echo ERROR: VCVARSALL is set but does not exist:
+    echo        "%VCVARSALL%"
+    echo        Update VCVARSALL or unset it so this script can auto-detect Visual Studio.
+    exit /b 1
+)
+
+if defined VSINSTALLDIR (
+    if exist "%VSINSTALLDIR%\VC\Auxiliary\Build\vcvarsall.bat" (
+        set "VCVARSALL=%VSINSTALLDIR%\VC\Auxiliary\Build\vcvarsall.bat"
+        exit /b 0
+    )
+)
+
+set "VSWHERE="
+if exist "%ProgramFiles(x86)%\Microsoft Visual Studio\Installer\vswhere.exe" (
+    set "VSWHERE=%ProgramFiles(x86)%\Microsoft Visual Studio\Installer\vswhere.exe"
+) else if exist "%ProgramFiles%\Microsoft Visual Studio\Installer\vswhere.exe" (
+    set "VSWHERE=%ProgramFiles%\Microsoft Visual Studio\Installer\vswhere.exe"
+) else (
+    for %%I in (vswhere.exe) do set "VSWHERE=%%~$PATH:I"
+)
+
+if defined VSWHERE (
+    for /f "usebackq delims=" %%I in (`"%VSWHERE%" -latest -products * -requires Microsoft.VisualStudio.Component.VC.Tools.ARM64 -property installationPath`) do (
+        if exist "%%I\VC\Auxiliary\Build\vcvarsall.bat" (
+            set "VCVARSALL=%%I\VC\Auxiliary\Build\vcvarsall.bat"
+            exit /b 0
+        )
+    )
+)
+
+echo ERROR: Could not find Visual Studio vcvarsall.bat with ARM64 C++ tools.
+echo        Install the C++ ARM64 build tools, or set VCVARSALL before running this script.
+echo        Example:
+echo        set "VCVARSALL=^<Visual Studio root^>\VC\Auxiliary\Build\vcvarsall.bat"
+exit /b 1
+
+
+:find_cuda
+echo Checking CUDA Toolkit...
+if not defined CUDA_PATH (
+    for /f "delims=" %%I in ('where nvcc.exe 2^>nul') do (
+        for %%J in ("%%~dpI..") do set "CUDA_PATH=%%~fJ"
+        goto :cuda_path_found
+    )
+)
+
+:cuda_path_found
+if not defined CUDA_PATH (
+    echo ERROR: CUDA_PATH is not set and nvcc.exe was not found on PATH.
+    echo        Install the native Windows ARM64 CUDA Toolkit, or set CUDA_PATH before running this script.
+    exit /b 1
+)
+
+if not exist "%CUDA_PATH%\include\cuda.h" (
+    echo ERROR: "%CUDA_PATH%\include\cuda.h" was not found.
+    echo        CUDA_PATH must point to the CUDA Toolkit root.
+    exit /b 1
+)
+
+if not exist "%CUDA_PATH%\lib\arm64\cuda.lib" (
+    echo ERROR: "%CUDA_PATH%\lib\arm64\cuda.lib" was not found.
+    echo        This script only builds native ARM64 PyCUDA and will not link against x64 CUDA libraries.
+    exit /b 1
+)
+
+set "CUDA_LIB_DIR=%CUDA_PATH%\lib\arm64"
+set "CUDA_ENABLE_CURAND=False"
+if exist "%CUDA_LIB_DIR%\curand.lib" set "CUDA_ENABLE_CURAND=True"
+
+rem CUDA 13+ Windows ARM64 toolkits keep runtime DLLs under bin\arm64 (and
+rem x64 under bin\x64). Prefer the native arm64 runtime dir so Python does not
+rem pick up an x64 curand/cudart DLL ("not a valid Win32 application").
+set "CUDA_BIN_DIR="
+if exist "%CUDA_PATH%\bin\arm64\curand64_10.dll" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin\arm64"
+) else if exist "%CUDA_PATH%\bin\arm64\cudart64_13.dll" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin\arm64"
+) else if exist "%CUDA_PATH%\bin\curand64_10.dll" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin"
+) else if exist "%CUDA_PATH%\bin\cudart64_13.dll" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin"
+) else if exist "%CUDA_PATH%\bin" (
+    set "CUDA_BIN_DIR=%CUDA_PATH%\bin"
+)
+
+if not defined CUDA_BIN_DIR (
+    echo ERROR: Could not locate a CUDA runtime bin directory under "%CUDA_PATH%\bin".
+    exit /b 1
+)
+
+set "PATH=%CUDA_BIN_DIR%;%CUDA_PATH%\bin;%PATH%"
+
+if not exist "%CUDA_PATH%\bin\nvcc.exe" (
+    echo WARNING: nvcc.exe was not found under CUDA_PATH\bin. PyCUDA driver APIs can still build, but pycuda.compiler will need nvcc later.
+)
+
+echo CUDA root : %CUDA_PATH%
+echo CUDA libs : %CUDA_LIB_DIR%
+echo CUDA bins : %CUDA_BIN_DIR%
+echo CURAND    : %CUDA_ENABLE_CURAND%
+exit /b 0
+
+:install_python_build_deps
+echo Installing Python build dependencies...
+"%PYTHON_EXE%" -m pip install --upgrade --prefer-binary setuptools wheel "numpy>=1.24" mako "platformdirs>=2.2.0" "pytools==%PYTOOLS_VERSION%"
+if errorlevel 1 (
+    echo ERROR: Failed to install Python build dependencies.
+    exit /b 1
+)
+exit /b 0
+
+:sync_pycuda_source
+echo Preparing PyCUDA source in "%PYCUDA_SOURCE_DIR%"...
+if not exist "%PYCUDA_BUILD_ROOT%" mkdir "%PYCUDA_BUILD_ROOT%"
+if errorlevel 1 (
+    echo ERROR: Could not create "%PYCUDA_BUILD_ROOT%".
+    exit /b 1
+)
+
+where git.exe >nul 2>nul
+if errorlevel 1 (
+    echo git.exe was not found on PATH. Using GitHub ZIP download fallback.
+    call :download_pycuda_source_without_git
+    exit /b !ERRORLEVEL!
+)
+
+if exist "%PYCUDA_SOURCE_DIR%\.git" (
+    git -C "%PYCUDA_SOURCE_DIR%" fetch --tags --depth 1 origin "%PYCUDA_TAG%"
+    if errorlevel 1 (
+        echo WARNING: Git fetch failed. Trying GitHub ZIP download fallback.
+        call :download_pycuda_source_without_git
+        exit /b !ERRORLEVEL!
+    )
+    git -C "%PYCUDA_SOURCE_DIR%" checkout --detach "%PYCUDA_TAG%"
+    if errorlevel 1 (
+        echo WARNING: Git checkout failed. Trying GitHub ZIP download fallback.
+        call :download_pycuda_source_without_git
+        exit /b !ERRORLEVEL!
+    )
+) else (
+    git clone --branch "%PYCUDA_TAG%" --depth 1 --recurse-submodules "%PYCUDA_REPO%" "%PYCUDA_SOURCE_DIR%"
+    if errorlevel 1 (
+        echo WARNING: Git clone failed. Trying GitHub ZIP download fallback.
+        call :download_pycuda_source_without_git
+        exit /b !ERRORLEVEL!
+    )
+)
+
+git -C "%PYCUDA_SOURCE_DIR%" submodule update --init --recursive
+if errorlevel 1 (
+    echo WARNING: Git submodule update failed. Trying GitHub ZIP download fallback.
+    call :download_pycuda_source_without_git
+    exit /b !ERRORLEVEL!
+)
+exit /b 0
+
+:download_pycuda_source_without_git
+where curl.exe >nul 2>nul
+if errorlevel 1 (
+    echo ERROR: git.exe is unavailable and curl.exe was not found.
+    echo        Install Git for Windows, or make curl.exe available on PATH.
+    exit /b 1
+)
+
+where tar.exe >nul 2>nul
+if errorlevel 1 (
+    echo ERROR: git.exe is unavailable and tar.exe was not found.
+    echo        Install Git for Windows, or make tar.exe available on PATH.
+    exit /b 1
+)
+
+call :download_and_extract_zip "https://github.com/inducer/pycuda/archive/%PYCUDA_COMMIT%.zip" "%PYCUDA_SOURCE_DIR%" "pycuda-"
+if errorlevel 1 exit /b 1
+
+call :download_and_extract_zip "https://github.com/inducer/bpl-subset/archive/%BPL_SUBSET_COMMIT%.zip" "%PYCUDA_SOURCE_DIR%\bpl-subset" "bpl-subset-"
+if errorlevel 1 exit /b 1
+
+call :download_and_extract_zip "https://github.com/inducer/compyte/archive/%COMPYTE_COMMIT%.zip" "%PYCUDA_SOURCE_DIR%\pycuda\compyte" "compyte-"
+if errorlevel 1 exit /b 1
+
+if not exist "%PYCUDA_SOURCE_DIR%\setup.py" (
+    echo ERROR: PyCUDA setup.py was not found after ZIP download.
+    exit /b 1
+)
+
+if not exist "%PYCUDA_SOURCE_DIR%\bpl-subset\bpl_subset\boost\python.hpp" (
+    echo ERROR: bpl-subset was not expanded correctly.
+    exit /b 1
+)
+
+if not exist "%PYCUDA_SOURCE_DIR%\pycuda\compyte\array.py" (
+    echo ERROR: compyte was not expanded correctly.
+    exit /b 1
+)
+
+exit /b 0
+
+:download_and_extract_zip
+set "ZIP_URL=%~1"
+set "ZIP_DEST=%~2"
+set "ZIP_PREFIX=%~3"
+set "ZIP_TEMP=%PYCUDA_BUILD_ROOT%\_zip_%RANDOM%_%RANDOM%"
+set "ZIP_FILE=%ZIP_TEMP%\source.zip"
+set "ZIP_EXTRACT_DIR=%ZIP_TEMP%\extract"
+set "ZIP_EXTRACTED="
+
+echo Downloading %ZIP_URL%
+if not exist "%ZIP_TEMP%" mkdir "%ZIP_TEMP%"
+if errorlevel 1 exit /b 1
+mkdir "%ZIP_EXTRACT_DIR%"
+if errorlevel 1 exit /b 1
+
+curl.exe -L --fail --retry 3 -o "%ZIP_FILE%" "%ZIP_URL%"
+if errorlevel 1 (
+    echo ERROR: Failed to download %ZIP_URL%
+    exit /b 1
+)
+
+tar.exe -xf "%ZIP_FILE%" -C "%ZIP_EXTRACT_DIR%"
+if errorlevel 1 (
+    echo ERROR: Failed to extract "%ZIP_FILE%".
+    exit /b 1
+)
+
+for /d %%D in ("%ZIP_EXTRACT_DIR%\%ZIP_PREFIX%*") do set "ZIP_EXTRACTED=%%~fD"
+if not defined ZIP_EXTRACTED (
+    echo ERROR: Could not find extracted folder matching "%ZIP_PREFIX%*" from %ZIP_URL%
+    exit /b 1
+)
+
+if not exist "%ZIP_DEST%" mkdir "%ZIP_DEST%"
+if errorlevel 1 exit /b 1
+
+robocopy "%ZIP_EXTRACTED%" "%ZIP_DEST%" /E >nul
+set "ROBOCOPY_STATUS=!ERRORLEVEL!"
+
+rmdir /s /q "%ZIP_TEMP%" >nul 2>nul
+
+if !ROBOCOPY_STATUS! GEQ 8 (
+    echo ERROR: Failed to copy extracted files into "%ZIP_DEST%".
+    exit /b !ROBOCOPY_STATUS!
+)
+
+exit /b 0
+
+:write_siteconf
+echo Writing PyCUDA siteconf.py...
+(
+    echo # Generated by setup_pycuda_win_arm64.bat.
+    echo BOOST_INC_DIR = []
+    echo BOOST_LIB_DIR = []
+    echo BOOST_COMPILER = "msvc"
+    echo USE_SHIPPED_BOOST = True
+    echo BOOST_PYTHON_LIBNAME = []
+    echo BOOST_THREAD_LIBNAME = []
+    echo CUDA_TRACE = False
+    echo CUDA_ROOT = r"%CUDA_PATH%"
+    echo CUDA_PRETEND_VERSION = None
+    echo CUDA_INC_DIR = [r"%CUDA_PATH%\include"]
+    echo CUDA_ENABLE_GL = False
+    echo CUDA_ENABLE_CURAND = %CUDA_ENABLE_CURAND%
+    echo CUDADRV_LIB_DIR = [r"%CUDA_LIB_DIR%"]
+    echo CUDADRV_LIBNAME = ["cuda"]
+    echo CUDART_LIB_DIR = [r"%CUDA_LIB_DIR%"]
+    echo CUDART_LIBNAME = ["cudart"]
+    echo CURAND_LIB_DIR = [r"%CUDA_LIB_DIR%"]
+    echo CURAND_LIBNAME = ["curand"]
+    echo CXXFLAGS = ["/EHsc", "/D_LITTLE_ENDIAN=1"]
+    echo LDFLAGS = ["/FORCE", "/MANIFEST:NO"]
+) > "%PYCUDA_SOURCE_DIR%\siteconf.py"
+if errorlevel 1 (
+    echo ERROR: Failed to write PyCUDA siteconf.py.
+    exit /b 1
+)
+exit /b 0
+
+:install_pycuda
+echo Building and installing PyCUDA %PYCUDA_VERSION%...
+pushd "%PYCUDA_SOURCE_DIR%"
+if errorlevel 1 exit /b 1
+
+set "DISTUTILS_USE_SDK=1"
+set "MSSdk=1"
+set "CL=/D_LITTLE_ENDIAN=1 %CL%"
+
+"%PYTHON_EXE%" -m pip install --no-build-isolation --no-deps --no-cache-dir --force-reinstall -v .
+set "PIP_STATUS=%ERRORLEVEL%"
+popd
+
+if not "%PIP_STATUS%"=="0" (
+    echo ERROR: PyCUDA build/install failed.
+    exit /b %PIP_STATUS%
+)
+exit /b 0
+
+:install_cuda_dll_path_hook
+rem Python 3.8+ on Windows does not search PATH for extension-module DLL
+rem dependencies. Register the native CUDA runtime dir via a site hook so
+rem "import pycuda" finds curand/cudart without manual add_dll_directory.
+echo Installing CUDA DLL search path hook for "%CUDA_BIN_DIR%"...
+
+if not defined CUDA_BIN_DIR (
+    echo ERROR: CUDA_BIN_DIR is not set.
+    exit /b 1
+)
+
+if not exist "%CUDA_BIN_DIR%" (
+    echo ERROR: CUDA_BIN_DIR does not exist: "%CUDA_BIN_DIR%"
+    exit /b 1
+)
+
+set "PYCUDA_SITE_DIR="
+set "PYCUDA_SITE_FILE=%PYCUDA_BUILD_ROOT%\pycuda_site_dir.txt"
+"%PYTHON_EXE%" -c "import sysconfig; print(sysconfig.get_path('purelib'))" > "%PYCUDA_SITE_FILE%"
+if errorlevel 1 (
+    echo ERROR: Could not query Python site-packages directory.
+    exit /b 1
+)
+set /p PYCUDA_SITE_DIR=<"%PYCUDA_SITE_FILE%"
+del /q "%PYCUDA_SITE_FILE%" >nul 2>nul
+if not defined PYCUDA_SITE_DIR (
+    echo ERROR: Could not determine Python site-packages directory.
+    exit /b 1
+)
+
+if not exist "%PYCUDA_SITE_DIR%" (
+    echo ERROR: Python site-packages directory does not exist: "%PYCUDA_SITE_DIR%"
+    exit /b 1
+)
+
+set "PYCUDA_DLL_HOOK=%PYCUDA_SITE_DIR%\_pycuda_cuda_dll_path.py"
+set "PYCUDA_DLL_PTH=%PYCUDA_SITE_DIR%\_pycuda_cuda_dll_path.pth"
+
+> "%PYCUDA_DLL_HOOK%" echo import os
+>> "%PYCUDA_DLL_HOOK%" echo from pathlib import Path
+>> "%PYCUDA_DLL_HOOK%" echo(
+>> "%PYCUDA_DLL_HOOK%" echo _CUDA_BIN = Path(r'''%CUDA_BIN_DIR%''')
+>> "%PYCUDA_DLL_HOOK%" echo _DLL_DIR_HANDLES = []
+>> "%PYCUDA_DLL_HOOK%" echo(
+>> "%PYCUDA_DLL_HOOK%" echo def _add():
+>> "%PYCUDA_DLL_HOOK%" echo     if not _CUDA_BIN.is_dir():
+>> "%PYCUDA_DLL_HOOK%" echo         return
+>> "%PYCUDA_DLL_HOOK%" echo     if hasattr(os, "add_dll_directory"):
+>> "%PYCUDA_DLL_HOOK%" echo         _DLL_DIR_HANDLES.append(os.add_dll_directory(str(_CUDA_BIN)))
+>> "%PYCUDA_DLL_HOOK%" echo     path = os.environ.get("PATH", "")
+>> "%PYCUDA_DLL_HOOK%" echo     prefix = str(_CUDA_BIN)
+>> "%PYCUDA_DLL_HOOK%" echo     parts = [p.lower() for p in path.split(os.pathsep)]
+>> "%PYCUDA_DLL_HOOK%" echo     if prefix.lower() not in parts:
+>> "%PYCUDA_DLL_HOOK%" echo         os.environ["PATH"] = prefix + os.pathsep + path
+>> "%PYCUDA_DLL_HOOK%" echo(
+>> "%PYCUDA_DLL_HOOK%" echo _add()
+
+> "%PYCUDA_DLL_PTH%" echo import _pycuda_cuda_dll_path
+
+if errorlevel 1 (
+    echo ERROR: Failed to install CUDA DLL search path hook.
+    exit /b 1
+)
+
+echo   site: %PYCUDA_SITE_DIR%
+echo   cuda bin: %CUDA_BIN_DIR%
+echo   wrote: _pycuda_cuda_dll_path.py, _pycuda_cuda_dll_path.pth
+exit /b 0
+
+:verify_pycuda
+echo Verifying PyCUDA import and CUDA driver initialization...
+"%PYTHON_EXE%" -u -X faulthandler -c "import platform, pycuda, pycuda.driver as cuda; print('  pycuda:', pycuda.VERSION_TEXT); print('  machine:', platform.machine()); cuda.init(); print('  cuda.init: ok'); print('  devices:', cuda.Device.count())"
+if errorlevel 1 (
+    echo ERROR: PyCUDA verification failed.
+    echo        If import still fails, ensure CUDA runtime DLLs are under "%CUDA_BIN_DIR%".
+    exit /b 1
+)
+exit /b 0
+
+:fail
+echo.
+echo PyCUDA setup failed.
+exit /b 1