lightning/requirements/base.txt

10 lines
202 B
Plaintext

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