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.

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

Does directml support multi-GPU training #406

Open
meyert11 opened this issue May 24, 2023 · 0 comments
Open

Does directml support multi-GPU training #406

meyert11 opened this issue May 24, 2023 · 0 comments

Comments

@meyert11
Copy link

My AMD GPUs have more VRAM than their NVIDIA counterparts, so I've been interested in trying directml lately. I was curious if directml, either tenosflow or pytorch versions, support mulit-GPU training? tensorflow.distribute mostly seems to lean on nccl, Nvidia specific, so I wasn't sure if directml had some special way it dealt with this for AMD cards? I saw someone asked a similar question in 2020 and you guys said it was part of your backlog (#70) , so I'm wondering if in 2023 that backlog has been filled...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@meyert11