-r requirements.txt

# For CUDA 12.*, you will need build onnxruntime-gpu from source and install the wheel. See README.md for detail.
# onnxruntime-gpu>=1.16.2

py3nvml

# The version of cuda-python shall be compatible with installed CUDA version.
# For demo of TensorRT excution provider and TensortRT.
cuda-python>=12.1.0

# For windows, cuda-python need the following
pywin32; platform_system == "Windows"

# For windows, run `conda install -c conda-forge nvtx` instead
nvtx; platform_system != "Windows"

# Please install PyTorch 2.1 or above for 12.1 using one of the following commands:
# pip3 install torch --index-url https://download.pytorch.org/whl/cu121

# Run the following command to install some extra packages for onnx graph optimization for TensorRT manually.
# pip3 install --upgrade polygraphy onnx-graphsurgeon --extra-index-url https://pypi.ngc.nvidia.com
