Install pytorch with Cuda 12.1

16

Thank you! I always used the CUDA 11.8 from the install matrix, but looks like it’s working now. If it helps others, here what I did:

I uninstalled torch through Termianl using “pip uninstall torch”

Then reinstalled using:
pip install torch torchvision --pre -f https://download.pytorch.org/whl/nightly/cu121/torch_nightly.html

This time it somehow installed correctly and now I get:

torch.cuda.is_available():
True

torch.version.cuda:
‘12.1’

pip list | findstr torch:
torch 2.1.0.dev20230604+cu121
torchvision 0.16.0.dev20230605+cu121

1 Like