lightning/requirements/extra.txt

12 lines
297 B
Plaintext
Raw Normal View History

# extended list of package dependencies to reach full functionality
matplotlib>3.1
horovod>=0.21.2 # no need to install with [pytorch] as pytorch is already installed
omegaconf>=2.0.1
torchtext>=0.7
2021-07-05 12:16:32 +00:00
onnx>=1.7.0
onnxruntime>=1.3.0
hydra-core>=1.0
jsonargparse[signatures]>=3.19.0
2021-07-05 12:16:32 +00:00
gcsfs>=2021.5.0