-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.10.0
torch==1.9.0+cpu
onnx==1.9.0
numpy
pycuda<2021.1
--extra-index-url https://pypi.ngc.nvidia.com
onnx-graphsurgeon
