-
Notifications
You must be signed in to change notification settings - Fork 9.5k
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
Update TP examples to align with tutorials #1243
Conversation
✅ Deploy Preview for pytorch-examples-preview canceled.
|
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.
Looks good! Had a minor suggestion.
not sure why the import fails... pls fix |
db7e1ce
to
9b3dd53
Compare
@@ -3,4 +3,4 @@ | |||
--pre | |||
--extra-index-url https://download.pytorch.org/whl/nightly/cu118 | |||
--extra-index-url https://download.pytorch.org/whl/nightly/cu121 | |||
torch >= 2.2.0.dev0; sys_platform == "linux" | |||
torch >= 2.3.0.dev0; sys_platform == "linux" |
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.
btw would this example work on T4 GPUs? Most of the distributed work here is still untested but there's no real reason for that IMO
Setting up distributed CI is as simple as runs-on: 4-core-ubuntu-gpu-t4
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.
Cool let me try this!
9ca0137
to
14f4f6e
Compare
as titled