tqdm
torch
torchvision
nibabel
einops
torchio
numpy==1.26.4
segmentation-models-pytorch==0.3.2
huggingface_hub
