8000 Fix multiprocessing with CUDA_VISIBLE_DEVICES seems to give the wrong device by fzyzcjy · Pull Request #149248 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

Fix multiprocessing with CUDA_VISIBLE_DEVICES seems to give the wrong device #149248

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.

Sign up for GitHub

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

Jump to bottom
Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Fix multiprocessing with CUDA_VISIBLE_DEVICES seems to give the wrong device #149248

wants to merge 1 commit into from
+13 −1

Conversation

fzyzcjy
Copy link
Contributor
@fzyzcjy fzyzcjy commented Mar 15, 2025

Fixes #149196

This is merely a proof-of-concept PR. I would like to hear a bit of feedback - is the direction acceptable - before working on it deeper.

Things that will be added if the direction of PR looks acceptable: Unit tests, caches, implement-in-C++ (to speedup), etc.

@pytorch-bot PyTorch Bot
Copy link
pytorch-bot bot commented Mar 15, 2025 •
edited
Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/149248

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit 98aef5c with merge base 1e37e5b (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@fzyzcjy fzyzcjy mentioned this pull request Mar 15, 2025
Open
@fzyzcjy
Copy link
Contributor Author
fzyzcjy commented Mar 15, 2025

@pytorchbot label "release notes: distributed (miscellaneous)"

@fzyzcjy fzyzcjy mentioned this pull request Mar 15, 2025
Merged
@albanD albanD added the triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module label Mar 17, 2025
@albanD
Copy link
Collaborator
albanD commented Mar 17, 2025

Let's discuss on the issue

@fzyzcjy fzyzcjy mentioned this pull request Mar 19, 2025
Merged
6 tasks
@github-actions GitHub Actions
Copy link
Contributor

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.

@github-actions github-actions bot added the Stale label May 16, 2025
@fzyzcjy
Copy link
Contributor Author
fzyzcjy commented May 16, 2025 •
edited
Loading

Hi is it possible to be merged?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
open source release notes: distributed (miscellaneous) Stale triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

(Will PR) Multiprocessing with CUDA_VISIBLE_DEVICES seems to give the wrong device
3 participants
@fzyzcjy @albanD @pytorchbot
0