numpy torch==1.13.1 torchvision matplotlib scipy h5py pandas kornia wandb pytorch-lightning==1.5.10 seaborn scikit-learn imageio imageio-ffmpeg gradio