torch
monai
antspyx
simpleitk
numpy
pandas
rich
scipy
scikit-learn
scikit-image
tensorboard
einops
pytest

[train]
nvidia-dali-cuda120
