What is the command to install pytorch with cuda 12.8?

Question

I notice on the website of pytorch (https://pytorch.org/get-started/locally/) there is a command for cuda 12.6 but not for the latest version of cuda which is 12.8.

Is there a command specifically for cuda 12.8?

Will the command for cuda 12.6 work if I'm using cuda 12.8?

Answer 1

The command to install the stable version of PyTorch (2.7.0) with CUDA 12.8 using pip on Linux is:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

Answer 2

as of now, pytorch which supports cuda 12.8 is not released yet.

but unofficial support released nightly version of it.

here are the commands to install it. so with this pytorch version you can use it on rtx 50XX. I've got 5080 and it works just fine.

pip install --pre torch==2.8.0.dev20250324+cu128 torchvision==0.22.0.dev20250325+cu128 torchaudio==2.6.0.dev20250325+cu128 --index-url https://download.pytorch.org/whl/nightly/cu128