lightning/requirements.txt

12 lines
270 B
Plaintext

# 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
pyDeprecate==0.1.1