Where can I find a version of pytorch which is suitable for CUDA12.8?

1

My os is win11 and gpu is 5090. First I tried CUDA12.8 but I can’t find a suitable version of pytorch so I have to download CUDA11.8 and use an old version of pytorch.
However, when I tried to train my network, it shows that my gpu with CUDA capability sm_120 is not compatible with the pytorch version.
I wonder where can I find a version of pytorch which is suitable for CUDA12.8?
Thanks!

2
3

Thanks! And according to the other replies, I wonder if torchvision and other libraries will support CUDA12.8 currently? Thanks!

4

I’m having the same issue and it hasn’t been resolved after following the reply above