torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 pytorch-lightning==1.3.5 opencv-python imageio ftfy regex tqdm altair protobuf timm tensorboardX matplotlib test-tube wandb Pillow