14 lines
459 B
Plaintext
14 lines
459 B
Plaintext
# 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.5
|
|
# onnx>=1.7.0
|
|
onnxruntime>=1.3.0
|
|
hydra-core>=1.0
|
|
# todo: when switch to standard package stream, drop `fairscale` from hard mocked docs libs
|
|
https://github.com/PyTorchLightning/fairscale/archive/pl_1.2.0.zip
|
|
jsonargparse[signatures]>=3.3.1
|
|
deepspeed>=0.3.13
|