-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Update fw_utils.py - support 2.0.1 container for DDP and Torch distri… #4127
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
…buted 2.0.1 exist and it works well with DDP and Torch Distributed, so allowing clients to use this combination.
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.
/bot run all
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
Codecov ReportPatch coverage:
Additional details and impacted files@@ Coverage Diff @@
## master #4127 +/- ##
==========================================
- Coverage 90.40% 90.35% -0.05%
==========================================
Files 990 312 -678
Lines 87099 28688 -58411
==========================================
- Hits 78739 25921 -52818
+ Misses 8360 2767 -5593
☔ View full report in Codecov by Sentry. |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
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.
/bot run pr
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
Update fw_utils.py - support 2.0.1 container for DDP and Torch distributed.
Issue #, if available:
Description of changes:
Update fw_utils.py - add support 2.0.1 container for DDP and Torch distruted.
2.0.1 exist and it works well with DDP and Torch Distributed, so allowing clients to use this combination.
see
763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker
in https://github.com/aws/deep-learning-containers/blob/master/available_images.mdTesting done: Changed fw_utils locally and saw it works fine with DDP and Torch Distributed.
Merge Checklist
Put an
x
in the boxes that apply. You can also fill these out after creating the PR. If you're unsure about any of them, don't hesitate to ask. We're here to help! This is simply a reminder of what we are going to look for before merging your pull request.General
Tests
unique_name_from_base
to create resource names in integ tests (if appropriate)By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.