huchukatoandDevin
6dc624c530
- 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. - Add pyproject.toml to identify the node uniquely in the Comfy registry. Generated with [Devin](https://devin.ai) Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com>
86 lines
2.7 KiB
Batchfile
86 lines
2.7 KiB
Batchfile
@echo off
|
|
REM Auto-install script for ComfyUI Upscaler 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 nvcc command
|
|
nvcc --version >nul 2>&1
|
|
if %errorlevel% equ 0 (
|
|
for /f "tokens=3" %%i in ('nvcc --version ^| find "release"') do (
|
|
set CUDA_VERSION=%%i
|
|
set CUDA_VERSION=!CUDA_VERSION:~0,-1!
|
|
)
|
|
if defined CUDA_VERSION (
|
|
echo Found CUDA version: !CUDA_VERSION!
|
|
)
|
|
)
|
|
|
|
REM Try CUDA_PATH if nvcc failed
|
|
if not defined CUDA_VERSION (
|
|
if defined CUDA_PATH (
|
|
if exist "%CUDA_PATH%\bin\nvcc.exe" (
|
|
for /f "tokens=3" %%i in ('"%CUDA_PATH%\bin\nvcc.exe" --version ^| find "release"') do (
|
|
set CUDA_VERSION=%%i
|
|
set CUDA_VERSION=!CUDA_VERSION:~0,-1!
|
|
)
|
|
if defined CUDA_VERSION (
|
|
echo Found CUDA version via CUDA_PATH: !CUDA_VERSION!
|
|
)
|
|
)
|
|
)
|
|
)
|
|
|
|
REM Try CUDA_HOME if still failed
|
|
if not defined CUDA_VERSION (
|
|
if defined CUDA_HOME (
|
|
if exist "%CUDA_HOME%\bin\nvcc.exe" (
|
|
for /f "tokens=3" %%i in ('"%CUDA_HOME%\bin\nvcc.exe" --version ^| find "release"') do (
|
|
set CUDA_VERSION=%%i
|
|
set CUDA_VERSION=!CUDA_VERSION:~0,-1!
|
|
)
|
|
if defined CUDA_VERSION (
|
|
echo Found CUDA version via CUDA_HOME: !CUDA_VERSION!
|
|
)
|
|
)
|
|
)
|
|
)
|
|
|
|
if not defined 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 MAJOR_VERSION=%%i
|
|
|
|
echo Installing requirements for CUDA !MAJOR_VERSION!...
|
|
|
|
if "!MAJOR_VERSION!"=="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 "!MAJOR_VERSION!"=="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 Upscaler TensorRT Auto node.
|
|
) else (
|
|
echo Installation failed!
|
|
)
|
|
pause
|