2020-03-25 11:46:27 +00:00
|
|
|
# extended list of package dependencies to reach full functionality
|
|
|
|
|
|
2020-12-02 16:18:14 +00:00
|
|
|
matplotlib>3.1
|
2021-02-09 17:25:57 +00:00
|
|
|
horovod>=0.21.2 # no need to install with [pytorch] as pytorch is already installed
|
2020-12-01 17:19:44 +00:00
|
|
|
omegaconf>=2.0.1
|
2021-03-02 15:38:02 +00:00
|
|
|
torchtext>=0.5
|
2021-03-15 11:07:23 +00:00
|
|
|
# onnx>=1.7.0
|
2020-08-05 09:32:53 +00:00
|
|
|
onnxruntime>=1.3.0
|
2020-11-27 00:00:56 +00:00
|
|
|
hydra-core>=1.0
|
2021-03-23 07:51:45 +00:00
|
|
|
# todo: when switch to standard package stream, drop `fairscale` from hard mocked docs libs
|
2021-02-16 19:43:02 +00:00
|
|
|
https://github.com/PyTorchLightning/fairscale/archive/pl_1.2.0.zip
|
2021-04-06 13:19:11 +00:00
|
|
|
jsonargparse[signatures]>=3.3.1
|
2021-03-30 17:39:02 +00:00
|
|
|
deepspeed>=0.3.13
|
