torch
monai
nibabel
torchvision
argparse
