pytorch-lightning==2.0.0.rc0
# no need to install with [pytorch] as pytorch is already installed
horovod>0.24.0, <=0.27.0
