apex==0.1+gitdb7007a.dtk2210
numpy==1.21.4
torch==1.10.0a0+git2040069.dtk2210
torchinfo==1.8.0