# GPU-enabled Python dependencies for RagGo (CUDA 12.1)
# Uses PyTorch CUDA index as the PRIMARY index to ensure the GPU wheel is
# selected over the CPU-only wheel on PyPI.
#
# If your system uses a different CUDA version, replace cu121 below:
#   cu118  = CUDA 11.8
#   cu121  = CUDA 12.1  (works on CUDA 12.x drivers)
#   cu124  = CUDA 12.4
#   cu126  = CUDA 12.6
#
# You can also override via env var before running the launcher:
#   set RAGGO_TORCH_CUDA=cu124
#
# IMPORTANT: --index-url replaces the default PyPI index for this install.
# --extra-index-url adds PyPI back as a fallback so all other packages are found.
--index-url https://download.pytorch.org/whl/cu121
--extra-index-url https://pypi.org/simple

# Base requirements
-r requirements.txt

# GPU Torch build (resolved from PyTorch CUDA index above)
torch
