Skip to content

Does directml support multi-GPU training #406

@meyert11

Description

@meyert11

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...

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions