PyTorch installation for CUDA 12 on win10 64 bit

Question

I need to install PyTorch on my PC, which has CUDA Version: 12.0.

The table for pytorch 2 in In pytorch site shows only CUDA 11.7 CUDA 11.8 as options.

previous versions of PyTorch doesn't mention CUDA 12 anywhere either.

What would be the most straightforward way to proceed?

Do I need to use an NGC container or build PyTorch from source?

thanks, Omer

Answer 1

One option is installing the nightly build that supports Python 3.11 and CUDA 12.1:

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