# Pinned CUDA 12 / TensorRT environment for local development and benchmarks.
#
# For a source checkout, install this file first and then install the package
# itself without re-resolving dependencies:
#
#   pip install -r requirements.txt
#   pip install -e . --no-deps
#
# Do NOT follow this with plain `pip install -e .`. A second dependency
# resolution pass can upgrade parts of the CUDA / ONNX Runtime stack to
# incompatible versions.
#
# Environment variables recommended for TensorRT:
#   export ORT_TENSORRT_CUDA_GRAPH_ENABLE=0   # Avoids stream capture conflicts in TRT 10.3+
#   export TRT_LOGGER_LEVEL=ERROR              # Suppress verbose TRT logs
#   export ORT_LOGGING_LEVEL=3                 # ORT ERROR level only

--extra-index-url https://pypi.nvidia.com

torch==2.5.1
numpy==2.4.1
spacy==3.8.2
spacy-transformers==1.3.9
thinc==8.3.10
transformers==4.41.2
onnx==1.20.1
cupy-cuda12x==13.6.0

# TensorRT Python bindings
tensorrt==10.15.1.29
tensorrt-cu12==10.15.1.29
tensorrt-cu12-bindings==10.15.1.29
tensorrt-cu12-libs==10.15.1.29

# NVIDIA's onnxruntime-gpu with TensorRT EP support
onnxruntime-gpu==1.23.2
