lightning/requirements.txt

11 lines
222 B
Plaintext

# the default package dependencies
numpy>=1.16.4
torch>=1.3,<1.8
future>=0.17.1 # required for builtins in setup.py
# pyyaml>=3.13
PyYAML>=5.1 # OmegaConf requirement >=5.1
tqdm>=4.41.0
fsspec>=0.8.0
tensorboard>=2.2.0