numpy==1.19.4; python_version<"3.7"
numpy==1.20.2; python_version>="3.7"
-f https://download.pytorch.org/whl/torch_stable.html
torch==1.9.0+cpu; (platform_machine=="x86_64" and sys.platform=="linux") or sys.platform=="win32"
torchvision==0.10.0+cpu; (platform_machine=="x86_64" and sys.platform=="linux") or sys.platform=="win32"
torch==1.9.0; platform_machine=="aarch64" and sys.platform=="linux"
torchvision==0.10.0; platform_machine=="aarch64" and sys.platform=="linux"
onnx==1.9.0
pycuda<2021.1
--extra-index-url https://pypi.ngc.nvidia.com
onnx-graphsurgeon
