requirements.txt

onnx>=1.14.0
tensorflow[and-cuda]==2.12.0; (python_version<"3.12" and platform_machine=="x86_64" and sys.platform=="linux" and python_version>="3.7")
tensorflow[and-cuda]==2.18.0; (python_version>="3.12" and platform_machine=="x86_64" and sys.platform=="linux")
onnxruntime==1.8.1; python_version<"3.10"
onnxruntime==1.12.1; python_version=="3.10"
-f https://download.pytorch.org/whl/cu113/torch_stable.html
torch==1.10.2+cu113; python_version<"3.10" and platform_machine=="x86_64" and sys.platform=="linux"
torch==1.10.2; python_version<"3.10" and platform_machine=="aarch64" and sys.platform=="linux"
torch==1.11.0+cu113; python_version=="3.10" and platform_machine=="x86_64" and sys.platform=="linux"
torch==1.11.0; python_version=="3.10" and platform_machine=="aarch64" and sys.platform=="linux"
torchvision==0.11.3; python_version<"3.10"
torchvision==0.12.0; python_version=="3.10"
Pillow
numpy
pycuda==2024.1
pytest
--extra-index-url https://pypi.ngc.nvidia.com
onnx-graphsurgeon