torch
torchvision
numpy
pillow
rasterio
albumentations
opencv-python
matplotlib
tqdm
scipy
einops
omegaconf
deepspeed; platform_system != "Windows"