2022-08-03 21:43:51 +00:00
|
|
|
# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
|
|
|
|
# in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment
|
|
|
|
|
2022-10-11 11:59:09 +00:00
|
|
|
colossalai>=0.1.10
|
2022-05-12 13:14:18 +00:00
|
|
|
fairscale>=0.4.5, <=0.4.6
|
2022-08-11 16:17:56 +00:00
|
|
|
deepspeed>=0.6.0, <=0.7.0
|
2022-05-12 13:14:18 +00:00
|
|
|
# no need to install with [pytorch] as pytorch is already installed
|
2022-06-22 11:23:15 +00:00
|
|
|
horovod>=0.21.2, !=0.24.0, <0.25.1
|
2022-05-12 13:14:18 +00:00
|
|
|
hivemind>=1.0.1, <=1.0.1; sys_platform == 'linux'
|