huchukatoandDevin
1427bdef9f
- Auto-install now only installs the matching TensorRT wheels instead of the multi-GB cuda-toolkit package from PyPI. - Use a single pip install -r call per requirements file for faster dependency resolution and better cache reuse. - Add install-success/failure markers to avoid repeating the install on every ComfyUI startup. - Fix pyproject.toml Repository and PublisherId to point to huchukato/ComfyUI-RIFE-TensorRT-Auto. Generated with [Devin](https://devin.ai) Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
66 lines
2.2 KiB
Batchfile
66 lines
2.2 KiB
Batchfile
@echo off
|
|
REM Auto-install script for ComfyUI RIFE TensorRT Auto (Windows)
|
|
REM Detects CUDA version and installs only the matching TensorRT wheels.
|
|
REM The NVIDIA CUDA Toolkit must already be installed on the system.
|
|
setlocal EnableDelayedExpansion
|
|
|
|
echo Detecting CUDA version...
|
|
|
|
REM Try to detect CUDA version
|
|
set CUDA_VERSION=
|
|
where nvcc >nul 2>nul
|
|
if %ERRORLEVEL% EQU 0 (
|
|
for /f "tokens=6 delims=, " %%i in ('nvcc --version ^| findstr "release"') do set CUDA_VERSION=%%i
|
|
echo Found CUDA version: %CUDA_VERSION%
|
|
) else if defined CUDA_PATH (
|
|
if exist "%CUDA_PATH%\bin\nvcc.exe" (
|
|
for /f "tokens=6 delims=, " %%i in ('"%CUDA_PATH%\bin\nvcc.exe" --version ^| findstr "release"') do set CUDA_VERSION=%%i
|
|
echo Found CUDA version via CUDA_PATH: %CUDA_VERSION%
|
|
)
|
|
) else if defined CUDA_HOME (
|
|
if exist "%CUDA_HOME%\bin\nvcc.exe" (
|
|
for /f "tokens=6 delims=, " %%i in ('"%CUDA_HOME%\bin\nvcc.exe" --version ^| findstr "release"') do set CUDA_VERSION=%%i
|
|
echo Found CUDA version via CUDA_HOME: %CUDA_VERSION%
|
|
)
|
|
)
|
|
|
|
if "%CUDA_VERSION%"=="" (
|
|
echo Could not detect CUDA version automatically.
|
|
echo Please ensure the NVIDIA CUDA Toolkit is installed and nvcc is in your PATH,
|
|
echo or set CUDA_PATH / CUDA_HOME.
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
REM Extract major version
|
|
for /f "tokens=1 delims=." %%i in ("%CUDA_VERSION%") do set CUDA_MAJOR=%%i
|
|
|
|
echo Installing requirements for CUDA %CUDA_MAJOR%...
|
|
|
|
REM Install appropriate requirements based on CUDA version
|
|
if "%CUDA_MAJOR%"=="13" (
|
|
echo Using CUDA 13 TensorRT packages ^(RTX 50 series^)
|
|
python -m pip install --prefer-binary -r requirements.txt
|
|
python -m pip install --prefer-binary -r requirements_cu13.txt
|
|
) else if "%CUDA_MAJOR%"=="12" (
|
|
echo Using CUDA 12 TensorRT packages ^(RTX 30/40 series^)
|
|
python -m pip install --prefer-binary -r requirements.txt
|
|
python -m pip install --prefer-binary -r requirements_cu12.txt
|
|
) else (
|
|
echo Unsupported CUDA version: %CUDA_VERSION%
|
|
echo Supported versions: CUDA 12.x, CUDA 13.x
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
if %ERRORLEVEL% EQU 0 (
|
|
echo Installation completed successfully!
|
|
echo You can now use the ComfyUI RIFE TensorRT Auto node.
|
|
) else (
|
|
echo Installation failed!
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
pause
|