cuda-python==12.9.0
cupy-cuda12x
torch
--extra-index-url https://pypi.ngc.nvidia.com
polygraphy>=0.50.1
colored
numpy==1.26.4
--extra-index-url https://pypi.ngc.nvidia.com
onnx-graphsurgeon
pywin32; platform_system == "Windows"
pyyaml==6.0.3
requests==2.32.4
tqdm==4.66.4