torch >= 2.0.0
torchvision >= 0.15.1
horovod >= 0.28.1