numpy
--extra-index-url https://download.pytorch.org/whl/cu113
torch==1.11.0
--extra-index-url https://download.pytorch.org/whl/cu113
torchvision
pytorch_lightning==1.6.0
scipy==1.7.1
protobuf==3.19.0