torch
torchvision
numpy
scipy
pytorch-lightning
nibabel
tqdm
