ZLUDA icon indicating copy to clipboard operation
ZLUDA copied to clipboard

Published 20 hours ago • verified publisher icon vosen

Support for PyTorch on windows?

Open Simmer1234569 opened this issue 1 year ago • 4 comments

I have been waiting a long time for PyTorch support in windows, it’s something that I really need as dual booting to Ubuntu is annoying and ruins how I would’ve worked usually. I did read that there is limited PyTorch support but I was wondering if it would work regularly for all to.device(‘cuda’) lines or does it only work for specific functions?

Simmer1234569 avatar Feb 13 '24 14:02 Simmer1234569

pytorch use CuDNN wich is Cuda Deep Neural Network. You can't run anything related to CuDNN (almost). Everything related to Pytorch CUDA will mostly not work. Don't expect anything and you MUST stay in your dual boot environement.

AnnoyedDev avatar Feb 14 '24 11:02 AnnoyedDev

Any plans to further improve support is it just not possible?

Simmer1234569 avatar Feb 14 '24 14:02 Simmer1234569

Also do you know if ROCM pytorch in windows is possible as of now?

Simmer1234569 avatar Feb 14 '24 21:02 Simmer1234569

Also do you know if ROCM pytorch in windows is possible as of now?

It isn't. pytorch needs some ROCm features wich aren't available on Windows yet. those are most likely features used by Zluda too under the hood

DGdev91 avatar Feb 15 '24 23:02 DGdev91