torch
psutil
torchio
pytorch-lightning
