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>
71 lines
2.7 KiB
Bash
Executable File
71 lines
2.7 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# FALLBACK install script for ComfyUI RIFE TensorRT Auto (Linux/macOS)
|
|
#
|
|
# NOTE: This is a BACKUP script - auto-installation is recommended!
|
|
#
|
|
# PRIMARY METHOD (Recommended):
|
|
# 1. pip install -r requirements.txt
|
|
# 2. Restart ComfyUI
|
|
# 3. Node auto-detects CUDA and installs TensorRT automatically
|
|
#
|
|
# Use this script ONLY if auto-installation fails!
|
|
#
|
|
# The NVIDIA CUDA Toolkit must already be installed on the system.
|
|
# This script installs only the matching TensorRT wheels via pip.
|
|
#
|
|
|
|
echo "FALLBACK INSTALLATION - Use only if auto-installation fails!"
|
|
echo "Detecting CUDA version..."
|
|
|
|
# Try to detect CUDA version
|
|
CUDA_VERSION=""
|
|
if command -v nvcc &> /dev/null; then
|
|
CUDA_VERSION=$(nvcc --version | grep "release" | awk '{print $6}' | cut -c2-)
|
|
echo "✅ Found CUDA version: $CUDA_VERSION"
|
|
elif [ -n "$CUDA_PATH" ] && [ -f "$CUDA_PATH/bin/nvcc" ]; then
|
|
CUDA_VERSION=$("$CUDA_PATH/bin/nvcc" --version | grep "release" | awk '{print $6}' | cut -c2-)
|
|
echo "✅ Found CUDA version via CUDA_PATH: $CUDA_VERSION"
|
|
elif [ -n "$CUDA_HOME" ] && [ -f "$CUDA_HOME/bin/nvcc" ]; then
|
|
CUDA_VERSION=$("$CUDA_HOME/bin/nvcc" --version | grep "release" | awk '{print $6}' | cut -c2-)
|
|
echo "✅ Found CUDA version via CUDA_HOME: $CUDA_VERSION"
|
|
else
|
|
echo "⚠️ Could not detect CUDA version automatically"
|
|
echo "Please ensure CUDA is installed and nvcc is in your PATH"
|
|
echo "Or set CUDA_PATH or CUDA_HOME environment variables"
|
|
exit 1
|
|
fi
|
|
|
|
# Extract major version
|
|
CUDA_MAJOR=$(echo $CUDA_VERSION | cut -d. -f1)
|
|
|
|
echo "📦 Installing requirements for CUDA $CUDA_MAJOR..."
|
|
|
|
# Install appropriate requirements based on CUDA version
|
|
if [ "$CUDA_MAJOR" = "13" ]; then
|
|
echo "Installing CUDA 13 requirements (RTX 50 series)"
|
|
echo "Installing base dependencies + CUDA 13 TensorRT..."
|
|
python3 -m pip install --prefer-binary -r requirements.txt
|
|
echo "Installing CUDA 13 specific TensorRT packages..."
|
|
python3 -m pip install --prefer-binary -r requirements_cu13.txt
|
|
elif [ "$CUDA_MAJOR" = "12" ]; then
|
|
echo "Installing CUDA 12 requirements (RTX 30/40 series)"
|
|
echo "Installing base dependencies + CUDA 12 TensorRT..."
|
|
python3 -m pip install --prefer-binary -r requirements.txt
|
|
echo "Installing CUDA 12 specific TensorRT packages..."
|
|
python3 -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"
|
|
exit 1
|
|
fi
|
|
|
|
if [ $? -eq 0 ]; then
|
|
echo "Fallback installation completed!"
|
|
echo "You can now use ComfyUI RIFE TensorRT Auto node"
|
|
echo "In the future, try auto-installation by just installing requirements.txt"
|
|
else
|
|
echo "Installation failed!"
|
|
exit 1
|
|
fi
|