Skip to content
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

New blueprint for terraform/ec2-example: ML distributed training #179

Merged
merged 21 commits into from
Jan 11, 2024

Conversation

sfloresk
Copy link
Contributor

Description

This solution blueprint creates the infrastructure to run distributed training jobs using a Ray cluster and PyTorch. The Ray head node runs on a m5.xlarge instance, while the 2 workers run on g5.12xlarge instances

Motivation and Context

There are no GPU related blueprints currently the terraform/ec2-example

How Has This Been Tested?

  • I have tested and validated these changes using one or more of the provided examples/* projects
  • I have executed pre-commit run -a on my pull request

@sfloresk sfloresk requested a review from a team as a code owner December 12, 2023 16:24
Copy link
Contributor

@joozero joozero left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@sfloresk Thanks ahead for pulling request for ecs-blueprint. Can you reflect these comments that I added?

@sfloresk
Copy link
Contributor Author

@joozero thanks for the feedback. I have modified the blueprint as requested, please let me know if you need anything else

@joozero joozero merged commit 67350e6 into aws-ia:main Jan 11, 2024
17 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants