--extra-index-url https://download.pytorch.org/whl/cu128
--extra-index-url https://pypi.nvidia.com
numpy<2.0.0
torch==2.7.1+cu128
cuda-python<13.0
torchvision==0.22.1+cu128
torchaudio==2.7.1+cu128
tensorrt==10.12.0.36
tensorrt-cu12==10.12.0.36
onnx==1.18.0
onnxruntime==1.22.0
onnxruntime-gpu==1.22.0
onnxmltools==1.14.0