lightning/requirements/extra.txt

11 lines
301 B
Plaintext

# extended list of package dependencies to reach full functionality
matplotlib>3.1
horovod>=0.21.2,!=0.24.0 # no need to install with [pytorch] as pytorch is already installed
torchtext>=0.9.*
omegaconf>=2.0.5
hydra-core>=1.0.5
jsonargparse[signatures]>=4.7.1
gcsfs>=2021.5.0
rich>=10.2.2,!=10.15.*