torch==1.4.0 torchvision==0.5.0 numpy tqdm tensorboard tensorboardX ml-collections medpy SimpleITK scipy h5py