pytorch-gradcam==0.2.1 pytorch-lightning==1.8.6 pytube==15.0.0 torch==2.0.1 torchmetrics==1.0.3 torchsummary==1.5.1 torchvision==0.15.2