cupy-cuda116
torch
torchvision
torchaudio
torchmetrics
matplotlib
cornucopia
nibabel
