lightning/requirements/base.txt

10 lines
234 B
Plaintext

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