-f https://download.pytorch.org/whl/torch_stable.html
torchvision==0.5.0; platform_machine=="x86_64" and sys.platform=="linux"
torch==1.4.0+cpu; platform_machine=="x86_64" and sys.platform=="linux"
onnx==1.8.0
numpy
pycuda<2021.1
