search

vosen / ZLUDA

CUDA on AMD GPUs
Apache License 2.0
8.29k stars 486 forks source link

Support for PyTorch on windows? #75

Open Simmer1234569 opened 4 months ago

Simmer1234569 commented 4 months ago

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?

AnnoyedDev commented 4 months ago

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.

Simmer1234569 commented 4 months ago

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

Simmer1234569 commented 4 months ago

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

DGdev91 commented 4 months ago

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