numpy==1.19.2 tensorflow==2.4 torch==1.7.0 torchsummary==1.5.1 torchvision==0.8.1 matplotlib ipython