lightning/requirements/strategies.txt

mirror of https://github.com/Lightning-AI/lightning.git
fairscale>=0.4.5
deepspeed<0.6.0
horovod>=0.21.2,!=0.24.0  # no need to install with [pytorch] as pytorch is already installed
hivemind>=1.0.1; sys_platform == 'linux'
Merge pull request #12723 from PyTorchLightning/req/strategies Separate strategies' requirements 2022-05-04 14:06:02 +00:00			`fairscale>=0.4.5`
			`deepspeed<0.6.0`
			`horovod>=0.21.2,!=0.24.0 # no need to install with [pytorch] as pytorch is already installed`
[1/2] Collaborative Strategy (#12842) 2022-05-05 16:06:26 +00:00			`hivemind>=1.0.1; sys_platform == 'linux'`