lightning/requirements.txt

# the default package dependencies

numpy>=1.16.6
torch>=1.4
future>=0.17.1  # required for builtins in setup.py
# pyyaml>=3.13
PyYAML>=5.1, !=5.4.*  # OmegaConf requirement >=5.1
tqdm>=4.41.0
fsspec[http]>=0.8.1
tensorboard>=2.2.0
torchmetrics>=0.2.0