-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Update deepspeed and fairscale versions
#12860
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
for more information, see https://pre-commit.ci
deepspeed>=0.6.0deepspeed and fairscale versions
| python -c "fname = 'requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if 'horovod' not in line] ; open(fname, 'w').writelines(lines)" | ||
| pip install fairscale>=0.4.5 | ||
| pip install deepspeed>=0.6.0 | ||
| pip install "fairscale>=0.4.5" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
For people dealing with CI/CD:
The command here was not appropriate! It just redirects stdout from pip install fairscale into a file named =0.4.5 and leads to installing the wrong versions, so make sure to put "" when specifying versions :)
- pip install fairscale>=0.4.5
+ pip install "fairscale>=0.4.5"There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
yep, I know, it just creates a file 0.4.5 :D
This reverts commit cb5a0ce.
|
@carmocca @Borda Addressed @awaelchli's comment. Could you have another look? |
* Fix deepspeed installation * Adapt to deepspeed>=0.5.9 * Fix fairscale installation Co-authored-by: Akihiro Nitta <[email protected]>
* Fix deepspeed installation * Adapt to deepspeed>=0.5.9 * Fix fairscale installation Co-authored-by: Akihiro Nitta <[email protected]>
What does this PR do?
Part of #12373.
0.5.7to0.5.10 (as of now)and avoids using0.6.{0,1}due to DeepSpeed v1.6.1 issues with Pytorch Lightning Deep Speed 3 Offloading deepspeedai/DeepSpeed#18780.4.0to0.4.6 (as of now)>=0.5.9. The change in DeepSpeed API was introduced in Refactor ZeRO naming to reduce confusion deepspeedai/DeepSpeed#1607.Does your PR introduce any breaking changes? If yes, please list them.
None
Before submitting
PR review
Anyone in the community is welcome to review the PR.
Before you start reviewing, make sure you have read the review guidelines. In short, see the following bullet-list:
Did you have fun?
Make sure you had fun coding 🙃
cc @carmocca @akihironitta @Borda @SeanNaren @awaelchli @rohitgr7