setup_windows_cuda.bat

@echo off
echo ========================================
echo Advanced Audio Stem Separator Setup
echo Windows CUDA GPU Support Installation
echo ========================================
echo.

echo Checking Python version...
python --version
echo.

echo Creating virtual environment with Python 3.12...
python -m venv venv_windows
echo.

echo Activating virtual environment...
call venv_windows\Scripts\activate.bat
echo.

echo Upgrading pip...
python -m pip install --upgrade pip
echo.

echo Installing setuptools and wheel...
pip install setuptools wheel
echo.

echo Uninstalling any existing PyTorch...
pip uninstall torch torchvision torchaudio -y
echo.

echo Installing CUDA-enabled PyTorch for GPU acceleration...
echo This will enable NVIDIA GPU detection and acceleration
pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0+cu118 --index-url https://download.pytorch.org/whl/cu118
echo.

echo Installing other dependencies...
pip install -r requirements.txt
echo.

echo Testing CUDA availability...
python -c "import torch; print('PyTorch version:', torch.__version__); print('CUDA available:', torch.cuda.is_available()); print('GPU count:', torch.cuda.device_count() if torch.cuda.is_available() else 0); print('GPU name:', torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'N/A')"
echo.

echo ========================================
echo Setup complete!
echo.
echo To use the application:
echo 1. Activate the environment: venv_windows\Scripts\activate.bat
echo 2. Run the application: python main.py
echo ========================================
echo.
echo Press any key to exit...
pause > nul