-
Notifications
You must be signed in to change notification settings - Fork 213
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
add deepspeed example #610
base: master
Are you sure you want to change the base?
Conversation
[APPROVALNOTIFIER] This PR is NOT APPROVED This pull-request has been approved by: The full list of commands accepted by this bot can be found here.
Needs approval from an approver in each of these files:
Approvers can indicate their approval by writing |
template: | ||
spec: | ||
containers: | ||
- image: registry.cn-beijing.aliyuncs.com/acs/deepspeed:hello-deepspeed |
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.
It would be useful to check in the Dockerfile as well.
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.
You'er right. This demo is intend to be used to test the up coming feature, I'll manage it later.
@kuizhiqing do you have a suggested place to run / test a setup like this? I've been trying to get just 3 nodes each with a single GPU on Google Cloud and I never get the allocation. I was able to get a single node with one GPU last week but it felt like luck. Will it work without GPU? |
It looks like it defaults to CPU, but it's not clear to me how communication is setup. Is it just using a shared volume at |
Hi @vsoch , actually, I've test it in the production environment of my affiliation. I've no idea where to run it in an open platform. The workspace path of the example contains materials from https://github.com/microsoft/DeepSpeedExamples/tree/master/training/HelloDeepSpeed, it do not involve the communication process. The communication setup by pdsh with the hostfile provided by the mpi-operator. For the CPU version, I'm afraid I cannot provide more information since I did not work with it. |
There would need to be some variant of an MPI run in there, and the communication (with the MPI operator) would happen via ssh bootstrap and then targeting that hostfile (which if I remember is an envar). I'm trying to understand where that logic is here. |
No description provided.