Skip to content

Pytorch DirectML integration in Pytorch Lightning #211

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
sto1 opened this issue Mar 10, 2022 · 5 comments
Open

Pytorch DirectML integration in Pytorch Lightning #211

sto1 opened this issue Mar 10, 2022 · 5 comments
Labels
pytorch-directml Issues in PyTorch when using its DirectML backend

Comments

@sto1
Copy link

sto1 commented Mar 10, 2022

Is there a plan to integrate DirectML into the Pytorch Lightning project?
This could broaden the base for DirectML fast.
@fdwr fdwr added the pytorch-directml Issues in PyTorch when using its DirectML backend label Apr 18, 2022
@Adele101
Copy link

Adele101 commented Aug 9, 2022

Hi @sto1,

Thank you for the great request, we will consider this for an upcoming release of pytorch-directml, and will update this thread when more information is available.

@wuzhican
Copy link

is there any update?

@arrrmin
Copy link

arrrmin commented Mar 13, 2023

I would also welcome an update.

@trupljan
Copy link

trupljan commented Aug 1, 2023 •
edited
Loading

I started implementing a prototype of the integration as a hobby project: Lightning-AI/pytorch-lightning#18188

Current prototype status:
Lightning-AI/pytorch-lightning@master...trupljan:lightning:master

@zhiyuan1i
Copy link

Maybe it can be down as a device plugin, I opened a new pr related to this: Lightning-AI/pytorch-lightning#20349

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
pytorch-directml Issues in PyTorch when using its DirectML backend
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
7 participants
@arrrmin @fdwr @sto1 @trupljan @zhiyuan1i @Adele101 @wuzhican