No support for Blackwell GPUs on cu128

[AlargerCorgi]

🚀 The feature

Trying to use nightly pyTorch builds which use cu128 on Blackwell NVIDIA GPUs.

"torchvision 0.22.0.dev20250316+cu128 requires torch==2.8.0.dev20250315+cu128, but you have torch 2.6.0 which is incompatible."

Motivation, pitch

NVIDIA's new GPUs use new cu128, so things like torchaudio/torchvision should be updated to 2.8.0 along with various other things.

Alternatives

No response

Additional context

No response

[ChatsideFires]

You are not doing something right - on 'install pytorch locally,' on their website, you can set up the little box to generate your command to install all the cu128 versions. I've been using them on the 4090 since they launched. But the issue really in this instance for torchaudio specifically, is that nightly builds ceased at the end of January when 2.6.0 became the current main release. I don't know if it got folded into torch itself or what, but it breaks a lot of compatibility.